QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#712699#8340. 3 SumCCSU_YZT#WA 358ms15196kbC++201.6kb2024-11-05 16:42:582024-11-05 16:42:59

Judging History

你现在查看的是最新测评结果

  • [2024-11-05 16:42:59]
  • 评测
  • 测评结果:WA
  • 用时:358ms
  • 内存:15196kb
  • [2024-11-05 16:42:58]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#define ll long long
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

bool isprime(int n) {
    if (n <= 1)
        return false;
    for (int i = 2; i * i <= n; i++) {
        if (n % i == 0)
            return false;
    }
    return true;
}
int findPrime(int n) {
    while (!isprime(n)) n++;
    return n;
}

void solve() {
    ll n, k;
    cin>>n>>k;
    vector<ll> a(n+1);
    vector<string> s(n+1);
    for (int i = 1; i <= n; i++) {
        cin>>s[i];
    }
    int M = 9e8, M1 = 1e8;
    int mod = findPrime(rng() % M + M1);
    set<ll> st;
    ll ans = 0;
    for (int t = 0; t <= 3; t++) {
        ll target = 0;
        for (int i = 1; i <= k; i++) {
            target = target * 10 + 9;
            target %= mod;
        }
        target = target * t % mod;
        st.insert(target);
    }

    for (int i = 1; i <= n; i++) {
        ll val = 0;
        for (auto ch : s[i]) {
            val = val * 10 % mod + ch - '0';
            val %= mod;
        }
        a[i] = val;
    }
    for (int i = 1; i <= n; i++) {
        for (int j = i; j <= n; j++) {
            for (int l = j; l <= n; l++) {
                if (st.count((a[i] + a[j] + a[l]) % mod)) {
                    ans++;
                }
            }
        }
    }
    cout<<ans<<'\n';
}

int main() {
    if (ifstream("test.in")) {
        freopen("test.in", "r", stdin);
        freopen("test.out", "w", stdout);
    }
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T = 1;
    // cin>>T;
    while (T--) {
        solve();
    }
    return 0;
} 

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3552kb

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 170ms
memory: 4404kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: 0
Accepted
time: 358ms
memory: 15196kb

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

0

result:

ok 1 number(s): "0"

Test #4:

score: -100
Wrong Answer
time: 205ms
memory: 15184kb

input:

500 1
751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...

output:

1

result:

wrong answer 1st numbers differ - expected: '2327631', found: '1'