QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#819311 | #8340. 3 Sum | piaoyun# | TL | 426ms | 114792kb | C++20 | 4.1kb | 2024-12-18 14:58:33 | 2024-12-18 14:58:42 |
Judging History
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