QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#668458 | #8340. 3 Sum | kjhhjki | WA | 79ms | 3864kb | C++20 | 1.9kb | 2024-10-23 14:30:10 | 2024-10-23 14:30:18 |
Judging History
answer
#include <bits/stdc++.h>
using i64 = long long;
using u32 = unsigned int;
using u64 = unsigned long long;
using f64 = long double;
template<typename T, typename U>
void chkmax(T &a, U b) { if(a < b) a = b; }
template<typename T, typename U>
void chkmin(T &a, U b) { if(a > b) a = b; }
// constexpr int P = 998244353;
constexpr u32 P = (17u << 27) + 1;
void init() { }
void solve()
{
int n, k;
std::cin >> n >> k;
u64 hk = 9;
std::vector<u64> hsh(n + 1), hp(k, 1);
for(int i = 1; i < k; ++i) {
hp[i] = hp[i - 1] * 10 % P;
hk = (hk + 9 * hp[i]) % P;
}
for(int i = 1; i <= n; ++i) {
std::string s;
std::cin >> s;
int m = s.length();
std::vector<int> c(k);
for(int j = 0; j < m; ++j) {
c[j % k] += s[m - j - 1] - '0';
}
// if(k == 1) {
// c[0] %= 9;
// } else {
// while(1) {
// for(int j = 0; j < k - 1; ++j) {
// c[j + 1] += c[j] / 10;
// c[j] %= 10;
// }
// if(c[k - 1] < 10) {
// break;
// }
// c[0] += c[k - 1] / 10;
// c[k - 1] %= 10;
// }
// }
for(int j = 0; j < k; ++j) {
hsh[i] = (hsh[i] + hp[j] * c[j]) % P;
}
}
int ans = 0;
for(int i = 1; i <= n; ++i) {
for(int j = i; j <= n; ++j) {
for(int k = j; k <= n; ++k) {
if(hsh[i] + hsh[j] + hsh[k] == 0 || hsh[i] + hsh[j] + hsh[k] == hk || (hsh[i] + hsh[j] + hsh[k]) % P == (hk * 2) % P) {
++ans;
}
}
}
}
std::cout << ans << '\n';
}
int main()
{
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
init();
int T = 1;
// std::cin >> T;
while(T --) solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3588kb
input:
4 1 0 1 10 17
output:
3
result:
ok 1 number(s): "3"
Test #2:
score: 0
Accepted
time: 36ms
memory: 3640kb
input:
500 859 7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 79ms
memory: 3756kb
input:
500 17336 11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...
output:
0
result:
ok 1 number(s): "0"
Test #4:
score: -100
Wrong Answer
time: 43ms
memory: 3864kb
input:
500 1 751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...
output:
0
result:
wrong answer 1st numbers differ - expected: '2327631', found: '0'