QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#379678 | #8340. 3 Sum | BUET_POTATOES# | WA | 102ms | 43576kb | C++20 | 3.4kb | 2024-04-06 18:15:08 | 2024-04-06 18:15:09 |
Judging History
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;
if(sum > 9){
sum -= 10;
carry = 1;
}
else{
carry = 0;
}
w[i] = sum;
}
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) == k)
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: 0ms
memory: 3612kb
input:
4 1 0 1 10 17
output:
3
result:
ok 1 number(s): "3"
Test #2:
score: 0
Accepted
time: 17ms
memory: 5708kb
input:
500 859 7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 102ms
memory: 43576kb
input:
500 17336 11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...
output:
0
result:
ok 1 number(s): "0"
Test #4:
score: 0
Accepted
time: 98ms
memory: 43372kb
input:
500 1 751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...
output:
2327631
result:
ok 1 number(s): "2327631"
Test #5:
score: 0
Accepted
time: 98ms
memory: 43352kb
input:
500 2 408542968136435277974575411503179002415404345446801430469044749372949272333801248935236224652806667129749035002588943020176263162139056819871274824302889304493205266143688886696157147111754418731401856424401766968832165255416237731963027205324149660112574729610391396555581935236134531799950318...
output:
212002
result:
ok 1 number(s): "212002"
Test #6:
score: -100
Wrong Answer
time: 87ms
memory: 43436kb
input:
500 11500 75411775796562109942642493394321873284995260953010112281856775261847503626737348402159485133662757032091519863427156582689971229143089317472838196453888261138079171290535429921921548971897026706656838415620603757605079012541561774699628665865662183868374645956694140356716037674688084770628...
output:
48
result:
wrong answer 1st numbers differ - expected: '7675', found: '48'