QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#665457#8340. 3 SumAI80WA 1ms3688kbC++201.9kb2024-10-22 13:01:462024-10-22 13:01:49

Judging History

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

  • [2024-10-22 13:01:49]
  • 评测
  • 测评结果:WA
  • 用时:1ms
  • 内存:3688kb
  • [2024-10-22 13:01:46]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define lowbit(x) (x & (-x))
typedef long long ll;
typedef pair<int,int> pii;
mt19937 rnd(time(0));
const int N = 2e4 + 7;
const int mod1 = 1145141819;
const int mod2 = 1e9 + 7;
// const int P = rnd() % mod;
string a[505];
pii h[505];
int ck[N];
int n , k;
string cal(string x) {
    string s;
    int n = x.length();
    for(int i = n - 1;i >= 0;i--) s += x[i];
    for(int i = 0;i < k;i++) ck[i] = 0;
    for(int i = 0;i < n;i++)  ck[i % k] += (s[i] - '0');
    int t = k,now = 0;
    while(t--) {
        if(ck[now] > 9) t = k;
        int res = ck[now] / 10;
        ck[now] %= 10;
        now = (now + 1) % k;
        ck[now] += res;
    }
    string ans = " ";
    for(int i = k - 1;i >= 0;i--) 
        ans += (char)(ck[i] + '0');
    return ans;
}
pii ha(string s) {
    int res1 = 0,res2 = 0;
    for(int i = 1;i <= k;i++) {
        res1 = (res1 * 10 + s[i] - '0') % mod1;
        res2 = (res2 * 10 + s[i] - '0') % mod2;
    }
    return {res1,res2};
}
void solve() {
    cin >> n >> k;
    for(int i = 1;i <= n;i++) cin >> a[i];
    for(int i = 1;i <= n;i++) a[i] = cal(a[i]);
    string k1 = " ", k2 = " ";
    for(int i = 1;i <= k;i++) k1 += '9';
    k2 += '1';
    for(int i = 2;i < k;i++) k2 += '9';
    k2 += '8';
    pii h1 = ha(k1),h2 = ha(k2),h0 = {0,0};
    for(int i = 1;i <= n;i++) h[i] = ha(a[i]);
    int ans = 0;
    for(int i = 1;i <= n;i++) {
        for(int j = i;j <= n;j++) {
            for(int k = j;k <= n;k++) {
                pii res = {(h[i].first + h[j].first + h[k].first) % mod1,(h[i].second + h[j].second + h[k].second) % mod2};
                if(res == h1 || res == h2 || res == h0) ans++;
            }
        }
    }
    cout << ans << "\n";
}
signed main() {
    ios::sync_with_stdio(0);
    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: 0
Wrong Answer
time: 1ms
memory: 3688kb

input:

4 1
0
1
10
17

output:

5

result:

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