QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#351709 | #8340. 3 Sum | karuna | AC ✓ | 101ms | 4016kb | C++20 | 2.2kb | 2024-03-12 13:12:00 | 2024-10-13 18:48:58 |
Judging History
answer
#include <bits/stdc++.h>
#define ff first
#define ss second
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> nd;
const ll P1 = 99999999999973ll;
const ll P2 = 999999999999989ll;
nd operator+(nd a, nd b) {
return {a.first + b.first, a.second + b.second};
}
int n, k;
int main() {
cin.tie(0); ios_base::sync_with_stdio(0);
cin >> n >> k;
vector<nd> V;
for (int i = 0; i < n; i++) {
string S;
cin >> S;
vector<int> a(k);
for (int j = 0; j < S.size(); j++) {
a[(S.size() - 1 - j) % k] += (S[j] - '0');
}
if (k == 1) {
a[0] %= 9;
}
else {
while (true) {
bool flag = false;
for (int i = 0; i < k; i++) {
if (a[i] >= 10) {
a[(i + 1) % k] += a[i] / 10;
a[i] %= 10;
flag = true;
}
}
if (!flag) break;
}
bool flag = true;
for (int i = 0; i < k; i++) {
if (a[i] != 9) flag = false;
}
if (flag) {
for (int &x : a) x = 0;
}
}
ll h1 = 0, h2 = 0;
for (int i = k - 1; i >= 0; i--) {
h1 = (10 * h1 + a[i]) % P1;
h2 = (10 * h2 + a[i]) % P2;
}
V.push_back({h1, h2});
}
sort(V.begin(), V.end());
vector<nd> A;
vector<int> B;
for (int i = 0; i < V.size(); i++) {
if (A.empty() || A.back() != V[i]) {
A.push_back(V[i]);
B.push_back(1);
}
else {
B.back()++;
}
}
int ans = 0;
for (int d = 0; d <= 18; d += 9) {
ll t1 = 0, t2 = 0;
for (int i = 0; i < k; i++) {
t1 = (10 * t1 + d) % P1;
t2 = (10 * t2 + d) % P2;
}
for (int s = 0; s < 3; s++) {
for (int q = 0; q < 3; q++) {
nd t = {t1 + s * P1, t2 + q * P2};
for (int i = 0; i < A.size(); i++) {
for (int j = i, k = A.size() - 1; j <= k; j++) {
while (k >= j && A[i] + A[j] + A[k] > t) --k;
if (k < j) break;
if (A[i] + A[j] + A[k] == t) {
if (i == j && j == k) {
ans += B[i] * (B[j] + 1) * (B[k] + 2) / 6;
}
else if (i == j) {
ans += B[i] * (B[j] + 1) * B[k] / 2;
}
else if (j == k) {
ans += B[i] * B[j] * (B[k] + 1) / 2;
}
else {
ans += B[i] * B[j] * B[k];
}
}
}
}
}
}
}
cout << ans << '\n';
}
这程序好像有点Bug,我给组数据试试?
详细
Test #1:
score: 100
Accepted
time: 0ms
memory: 3620kb
input:
4 1 0 1 10 17
output:
3
result:
ok 1 number(s): "3"
Test #2:
score: 0
Accepted
time: 12ms
memory: 3616kb
input:
500 859 7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 97ms
memory: 3984kb
input:
500 17336 11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...
output:
0
result:
ok 1 number(s): "0"
Test #4:
score: 0
Accepted
time: 36ms
memory: 3772kb
input:
500 1 751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...
output:
2327631
result:
ok 1 number(s): "2327631"
Test #5:
score: 0
Accepted
time: 36ms
memory: 3712kb
input:
500 2 408542968136435277974575411503179002415404345446801430469044749372949272333801248935236224652806667129749035002588943020176263162139056819871274824302889304493205266143688886696157147111754418731401856424401766968832165255416237731963027205324149660112574729610391396555581935236134531799950318...
output:
212002
result:
ok 1 number(s): "212002"
Test #6:
score: 0
Accepted
time: 96ms
memory: 3940kb
input:
500 11500 75411775796562109942642493394321873284995260953010112281856775261847503626737348402159485133662757032091519863427156582689971229143089317472838196453888261138079171290535429921921548971897026706656838415620603757605079012541561774699628665865662183868374645956694140356716037674688084770628...
output:
7675
result:
ok 1 number(s): "7675"
Test #7:
score: 0
Accepted
time: 101ms
memory: 4016kb
input:
500 11500 85355036663164764459816544518601485185320972076300982726542821424439713703229669576802138231401047471351087455159512255765325323540671792953715169122669767368905391325060775725733157611188832204902997772518104188947349204726490597030311894441123834099315122116302203972018409854605418988681...
output:
1070
result:
ok 1 number(s): "1070"
Test #8:
score: 0
Accepted
time: 1ms
memory: 3580kb
input:
1 11500 9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999...
output:
1
result:
ok 1 number(s): "1"
Extra Test:
score: 0
Extra Test Passed