QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#668405#8340. 3 Sumkjhhjki_fan#WA 87ms3752kbC++201.8kb2024-10-23 14:17:212024-10-23 14:17:24

Judging History

你现在查看的是最新测评结果

  • [2024-10-23 14:17:24]
  • 评测
  • 测评结果:WA
  • 用时:87ms
  • 内存:3752kb
  • [2024-10-23 14:17:21]
  • 提交

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';
        }
        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] == hk * 2) {
                    ++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: 3592kb

input:

4 1
0
1
10
17

output:

3

result:

ok 1 number(s): "3"

Test #2:

score: 0
Accepted
time: 14ms
memory: 3544kb

input:

500 859
7118711592236878297922359501613604144948355616986970837340677671376753603836852811886591300370143151943368529129749813118476151865844255212534355441611481420938483178075143062691345257288242460282715389758789648541099090735875617822348551942134616963557723055980260082230902505269975518146286...

output:

0

result:

ok 1 number(s): "0"

Test #3:

score: 0
Accepted
time: 87ms
memory: 3752kb

input:

500 17336
11871159223687829792235950161360414494835561698697083734067767137675360383685281188659130037014315194336852912974981311847615186584425521253435544161148142093848317807514306269134525728824246028271538975878964854109909073587561782234855194213461696355772305598026008223090250526997551814628...

output:

0

result:

ok 1 number(s): "0"

Test #4:

score: -100
Wrong Answer
time: 40ms
memory: 3668kb

input:

500 1
751324443898124078584847834484321089092662321556147445230263526014359393841194947303407593948729802551881289193716611867931891257925091769456350249725997883453296895094445731130479434019358742162771547784250401546380268386074363779242500860317042151185119666027858022664683818314351285215150806...

output:

2315119

result:

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