QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#712590#8340. 3 SumCCSU_YZT#WA 491ms15064kbC++201.7kb2024-11-05 16:17:552024-11-05 16:17:55

Judging History

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

  • [2024-11-05 16:17:55]
  • 评测
  • 测评结果:WA
  • 用时:491ms
  • 内存:15064kb
  • [2024-11-05 16:17:55]
  • 提交

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];
    }
    ll ans = 0;
    for (int t = 0; t <= 3; t++) {
        int M = 9e8, M1 = 1e8;
        int mod = findPrime(rng() % M + M1);
        ll target = 0;
        for (int i = 1; i <= k; i++) {
            target = target*10 + 9;
            target %= mod;
        }
        target = target * t % mod;
        for (int i = 1; i <= n; i++) {
            ll val = 0;
            for (auto ch : s[i]) {
                val = val * 10 + ch-'0';
                val %= mod;
            }
            a[i] = val;
        }
        for (int i = 1; i <= n; i++) {
            for (int j = i; j <= n; j++) {
                for (int k = j; k <= n; k++) {
                    if ((a[i] + a[j] + a[k])%mod == target) {
                        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: 1ms
memory: 3616kb

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 254ms
memory: 4120kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: -100
Wrong Answer
time: 491ms
memory: 15064kb

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

1

result:

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