QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#949551#8340. 3 Sumlyx1311WA 185ms13696kbC++202.3kb2025-03-24 02:06:362025-03-24 02:06:36

Judging History

This is the latest submission verdict.

  • [2025-03-24 02:06:36]
  • Judged
  • Verdict: WA
  • Time: 185ms
  • Memory: 13696kb
  • [2025-03-24 02:06:36]
  • Submitted

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 < std::min(j + K, len); k++) {
                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]++;
            }
            if(a[i][K - 1] >= 10) a[i][K - 1] -= 10, a[i][K]++;
            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;
}

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 3840kb

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 58ms
memory: 12452kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: 0
Accepted
time: 185ms
memory: 13668kb

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

0

result:

ok 1 number(s): "0"

Test #4:

score: 0
Accepted
time: 93ms
memory: 13600kb

input:

500 1
751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...

output:

2327631

result:

ok 1 number(s): "2327631"

Test #5:

score: 0
Accepted
time: 91ms
memory: 13696kb

input:

500 2
408542968136435277974575411503179002415404345446801430469044749372949272333801248935236224652806667129749035002588943020176263162139056819871274824302889304493205266143688886696157147111754418731401856424401766968832165255416237731963027205324149660112574729610391396555581935236134531799950318...

output:

212002

result:

ok 1 number(s): "212002"

Test #6:

score: -100
Wrong Answer
time: 161ms
memory: 13436kb

input:

500 11500
75411775796562109942642493394321873284995260953010112281856775261847503626737348402159485133662757032091519863427156582689971229143089317472838196453888261138079171290535429921921548971897026706656838415620603757605079012541561774699628665865662183868374645956694140356716037674688084770628...

output:

1

result:

wrong answer 1st numbers differ - expected: '7675', found: '1'