QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#379665#8340. 3 SumBUET_POTATOES#WA 106ms43576kbC++203.4kb2024-04-06 18:09:382024-04-06 18:09:39

Judging History

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

  • [2024-09-20 10:20:30]
  • hack成功,自动添加数据
  • (/hack/848)
  • [2024-04-06 18:09:39]
  • 评测
  • 测评结果:WA
  • 用时:106ms
  • 内存:43576kb
  • [2024-04-06 18:09:38]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
using ll = long long int;

const ll M1 = 1e9 + 9, M2 = 1e9 + 21;

vector<int> input[505];

void print(vector<int>& v){
    cout<<"> ";
    for(int i = (int) v.size() - 1; i>=0; i--){
        cout<<v[i];
    }
    cout<<endl;
}

void reduce(vector<int>& v, int k){
    if(v.size() > k){
        vector<int> w(v.begin(), v.begin()+k);

        int shift = k;
        auto getDigit = [&](int i) {
            int ii = shift + i;
            if(ii >= shift+k || ii >= v.size()) return 0;
            return v[ii];
        };

        while(shift < v.size()){
            int carry = 0;
            for(int i = 0; i < k; i++){
                int sum = w[i] + getDigit(i) + carry;
                w[i] = sum;
                if(sum > 9){
                    sum -= 10;
                    carry = 1;
                }
                else{
                    carry = 0;
                }
            }

            if(carry == 1){
                for(int i = 0; i<k; i++){
                    if(w[i] == 9){
                        w[i] = 0;
                    }
                    else{
                        w[i]++;
                        break;
                    }
                }
            }

            shift += k;
        }

        v = w;
    }

    if(count(v.begin(), v.end(), 9) == v.size())
        v = vector<int>{0};
    return;
}

pair<ll,ll> getHash(vector<int>& v){
    ll res1 = 0, res2 = 0;

    for(int x : v){
        res1 = (res1 * 10 + x) % M1;
        res2 = (res2 * 10 + x) % M2;
    }

    return make_pair(res1, res2);
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);

    int n, k;
    cin>>n>>k;

    for(int i = 0; i<n; i++){
        string s;
        cin>>s;
        for(int j = (int)s.size() - 1; j>=0; j--){
            input[i].push_back(s[j] - '0');
        }
    }

    for(int i = 0; i<n; i++){
        reduce(input[i], k);
    }

    vector<int> m(k, 9);

    vector<pair<ll, ll>> v;
    for(int i = 0; i<n; i++){
        v.emplace_back(getHash(input[i]));
//        cout<<v[i].first<<" - "<<v[i].second<<endl;
    }

    pair<ll, ll> target0 = make_pair(0LL, 0LL);
    pair<ll, ll> target = getHash(m);
    pair<ll, ll> target2 = make_pair(target.first * 2 % M1, target.second * 2 % M2);

//    cout<<target.first<<" - "<<target.second<<endl;
//    cout<<target2.first<<" - "<<target2.second<<endl;


    map<pair<ll, ll>, int> mp;
    int ans = 0;

    for(int i = 0; i<n; i++){
        mp[v[i]]++;

        for(int j = i; j<n; j++){
            auto x = make_pair((v[i].first + v[j].first) % M1, (v[i].second + v[j].second) % M2);

            auto cand0 = make_pair((M1 + target0.first - x.first) % M1, (M2 + target0.second - x.second) % M2);
            auto cand1 = make_pair((M1 + target.first - x.first) % M1, (M2 + target.second - x.second) % M2);
            auto cand2 = make_pair((M1 + target2.first - x.first) % M1, (M2 + target2.second - x.second) % M2);

            auto it = mp.find(cand0);
            if(it != mp.end()) ans += it->second;

            it = mp.find(cand1);
            if(it != mp.end()) ans += it->second;

            it = mp.find(cand2);
            if(it != mp.end()) ans += it->second;

//            cout<<ans<<" - "<<i<<" "<<j<<endl;
        }
    }

    cout<<ans<<endl;

}
/*
4 1
0
1
10
17
*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3780kb

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 19ms
memory: 5584kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: 0
Accepted
time: 106ms
memory: 43576kb

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

0

result:

ok 1 number(s): "0"

Test #4:

score: -100
Wrong Answer
time: 75ms
memory: 43432kb

input:

500 1
751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...

output:

0

result:

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