QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#949552 | #8340. 3 Sum | lyx1311 | WA | 189ms | 13704kb | C++20 | 2.2kb | 2025-03-24 02:13:27 | 2025-03-24 02:13:27 |
Judging History
answer
#include <bits/stdc++.h>
typedef unsigned long long ULL;
#define TIME std::chrono::system_clock::now().time_since_epoch().count()
template<typename Tp> Tp Rand(const Tp l, const Tp r) {
static std::mt19937_64 rng(TIME);
return Tp(double(rng()) / ULLONG_MAX * (r - l + 1)) + l;
}
const int M = 507, N = 2e4 + 7;
int n, K, cnt0 = 0, ans = 0; char a[M][N]; ULL P[2], s[M][2];
std::map<std::pair<ULL, ULL>, int> mp;
void solve(ULL m0, ULL m1) {
auto count = [&](int i, int j) {
ULL h0 = (m0 - s[i][0] - s[j][0] + P[0] * 2) % P[0];
ULL h1 = (m1 - s[i][1] - s[j][1] + P[1] * 2) % P[1];
return mp.count(std::make_pair(h0, h1)) ? mp[std::make_pair(h0, h1)] : 0;
};
int sum = 0;
for(int i = 1; i <= n; i++) {
mp[std::make_pair(s[i][0], s[i][1])]--;
for(int j = i + 1; j <= n; j++) {
mp[std::make_pair(s[j][0], s[j][1])]--;
sum += count(i, j);
mp[std::make_pair(s[j][0], s[j][1])]++;
}
mp[std::make_pair(s[i][0], s[i][1])]++;
ans += count(i, i);
}
ans += sum / 3;
}
int main() {
for(int i = 0; i < 2; i++) P[i] = Rand((ULL)5e18, (ULL)6e18);
scanf("%d%d", &n, &K);
for(int i = 1; i <= n; i++) {
scanf("%s", a[i]); int len = strlen(a[i]);
std::reverse(a[i], a[i] + len);
for(int j = 0; j < len; j++) a[i][j] -= '0';
for(int j = len / K * K; j; j -= K) do {
for(int k = j; k < j + K; k++) {
if(k < len) a[i][k - j] += a[i][k], a[i][k] = 0;
if(a[i][k - j] >= 10) a[i][k - j] -= 10, a[i][k - j + 1]++;
}
int k = K;
while(a[i][k] >= 10) a[i][k] -= 10, a[i][++k]++;
} while(a[i][j]);
bool all9 = 1;
for(int j = 0; j < K; j++) if(a[i][j] != 9) all9 = 0;
if(all9) for(int j = 0; j < K; j++) a[i][j] = 0;
for(int j = K; j; j--) s[i][0] = (s[i][0] * 10 + a[i][j - 1]) % P[0];
for(int j = K; j; j--) s[i][1] = (s[i][1] * 10 + a[i][j - 1]) % P[1];
mp[std::make_pair(s[i][0], s[i][1])]++;
}
ULL m0 = 0, m1 = 0;
for(int i = 0; i < K; i++) m0 = (m0 * 10 + 9) % P[0], m1 = (m1 * 10 + 9) % P[1];
solve(0, 0), solve(m0, m1), solve(m0 * 2 % P[0], m1 * 2 % P[1]);
return printf("%d\n", ans), 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3840kb
input:
4 1 0 1 10 17
output:
3
result:
ok 1 number(s): "3"
Test #2:
score: 0
Accepted
time: 57ms
memory: 12468kb
input:
500 859 7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 189ms
memory: 13552kb
input:
500 17336 11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...
output:
0
result:
ok 1 number(s): "0"
Test #4:
score: 0
Accepted
time: 85ms
memory: 13700kb
input:
500 1 751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...
output:
2327631
result:
ok 1 number(s): "2327631"
Test #5:
score: 0
Accepted
time: 94ms
memory: 13600kb
input:
500 2 408542968136435277974575411503179002415404345446801430469044749372949272333801248935236224652806667129749035002588943020176263162139056819871274824302889304493205266143688886696157147111754418731401856424401766968832165255416237731963027205324149660112574729610391396555581935236134531799950318...
output:
212002
result:
ok 1 number(s): "212002"
Test #6:
score: -100
Wrong Answer
time: 164ms
memory: 13704kb
input:
500 11500 75411775796562109942642493394321873284995260953010112281856775261847503626737348402159485133662757032091519863427156582689971229143089317472838196453888261138079171290535429921921548971897026706656838415620603757605079012541561774699628665865662183868374645956694140356716037674688084770628...
output:
1
result:
wrong answer 1st numbers differ - expected: '7675', found: '1'