QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#819311#8340. 3 Sumpiaoyun#TL 426ms114792kbC++204.1kb2024-12-18 14:58:332024-12-18 14:58:42

Judging History

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

  • [2024-12-18 14:58:42]
  • 评测
  • 测评结果:TL
  • 用时:426ms
  • 内存:114792kb
  • [2024-12-18 14:58:33]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
#define int long long
const long long MOD[5] = {998244353, 1000000000 + 7, 1000000000 + 9, 19260817, 1000000000 + 3};
const int MAXN = 1e6 + 10;
int N,M,K,Q;
int a[MAXN];
struct Num{
    int val[20010];
    int top = 0;
    vector<long long> hash;

    void read(string &str){
        top = str.size() - 1;
        for(int i = 0; i <= top; i++){
            val[i] = str[str.size() - 1 - i] - '0';
        }
    }

    void ModK(){
        for(int base = top / K * K; base; base -= K){
            while(1){
                int flag = 0;
                for(int k = 0; k < K; k++){
                    if(base + k > 20005) break;
                    val[base - K + k] += val[base + k];
                    if(val[base + k]) flag = 1;
                    val[base + k] = 0;
                }
                if(flag == 0) break;
                for(int k = 0; k < K; k++){
                    if(val[base - K + k] >= 10){
                        val[base - K + k] -= 10;
                        val[base - K + k + 1]++;
                    }
                }
            }
        }
        int not9 = 0;
        for(int k = 0; k < K; k++){
            if(val[k] != 9) not9 = 1;
        }
        if(!not9){
            for(int k = 0; k < K; k++){
                val[k] = 0;
            }
        }
        top = 0;
        for(int k = 0; k < K; k++){
            if(val[k]) top = k;
        }
    }

    void print(){
        for(int i = top; i >= 0; i--){
            cout << val[i];
        }
        cout << endl;
    }

    void getHash(){
        hash.clear();
        hash.resize(5, 0);
        for(int i = top; i >= 0; i--){
            for(int k = 0; k < 5; k++){
                hash[k] = (hash[k] * 10 + val[i]) % MOD[k];
            }
        }
    }
}nums[510];

string str[510];

vector<long long> add(vector<long long> &a, vector<long long> &b){
    vector<long long> ret(5,0);
    for(int k = 0; k < 5; k++){
        ret[k] = (a[k] + b[k]) % MOD[k];
    }
    return ret;
}

vector<long long> del(vector<long long> &a, vector<long long> &b){
    vector<long long> ret(5,0);
    for(int k = 0; k < 5; k++){
        ret[k] = (a[k] - b[k]);
        if(ret[k] < 0) ret[k] += MOD[k];
    }
    return ret;
}

map<vector<long long>, int> mp;

void prepare(){
	cin >> N >> K;
    for(int i = 1; i <= N; i++){
        cin >> str[i];
        nums[i].read(str[i]);
        nums[i].ModK();
        nums[i].getHash();
        //cout << nums[i].hash[0] << "?" << endl;
    }
    nums[501].top = K - 1;
    for(int i = 0; i <= K - 1; i++){
        nums[501].val[i] = 9;
    }
    nums[501].getHash();

    nums[502].top = K;
    for(int i = 1; i <= K - 1; i++){
        nums[502].val[i] = 9;
    }
    nums[502].val[K] = 1;
    nums[502].val[0] = 8;
    nums[502].getHash();

    nums[503].top = 0;
    nums[503].getHash();
    for(int i = 1; i <= N; i++){
        mp[nums[i].hash]++;
        //for(int k = 0; k < 5; k++) cout << nums[i].hash[k] << " ";
        //cout << endl;
    }
    // for(int t = 1; t <= 3; t++){
    //     for(int k = 0; k < 5; k++) cout << nums[200 + t].hash[k] << " ";
    //     cout << endl;
    // }
    int ans = 0;
    for(int i = 1; i <= N; i++){
        //mp[nums[i].hash]--;
        for(int j = i; j <= N; j++){
            //mp[nums[j].hash]--;
            for(int k = 1; k <= 3; k++){
                vector<long long> tar = nums[500 + k].hash;
                tar = del(tar, nums[i].hash);
                tar = del(tar, nums[j].hash);
                //for(int t = 0; t < 1; t++) cout << tar[t] << " ";
                //cout << endl;
                //cout << i << " " << j << " " << mp[tar] << endl;
                ans += mp[tar];
            }
            mp[nums[j].hash]--;
        }
        for(int j = i; j <= N; j++){
            mp[nums[j].hash]++;
        }
        mp[nums[i].hash]--;
    }
    cout << ans << '\n';
}

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T = 1;
	//cin >> T;
	while(T--){
		prepare();
	}
}



Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 426ms
memory: 114792kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: -100
Time Limit Exceeded

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

0

result: