QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#234374#6522. Digit Modeucup-team1198#AC ✓330ms3852kbC++202.9kb2023-11-01 16:42:422023-11-01 16:42:42

Judging History

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

  • [2023-11-01 16:42:42]
  • 评测
  • 测评结果:AC
  • 用时:330ms
  • 内存:3852kb
  • [2023-11-01 16:42:42]
  • 提交

answer

#include <bits/stdc++.h>

using namespace std;
#define ll long long
#define pii pair<int, int>
#define ld long double
#define all(a) (a).begin(), (a).end()

const int MOD = 1e9 + 7;

int add(int a, int b) {
    if (a + b < MOD)
        return a + b;
    return a + b - MOD;
}

int sub(int a, int b) {
    if (a >= b)
        return a - b;
    return a + MOD - b;
}


int mul(int a, int b) {
    return a * 1ll * b % MOD;
}

int pw(int a, int n) {
    int ans = 1;
    while (n) {
        if (n & 1)
            ans = mul(ans, a);
        a = mul(a, a);
        n >>= 1;
    }
    return ans;
}

const int MAXN = 55;
int C[MAXN][MAXN];

int get_cnt(int k, vector<int> mx_cnt) {
    vector<int> dp(k + 1, 0);
    vector<int> new_dp(k + 1, 0);
    dp[0] = 1;
    for (int x : mx_cnt) {
        fill(new_dp.begin(), new_dp.end(), 0);
        for (int j = 0; j <= x; ++j) {
            for (int i = 0; i + j <= k; ++i)
                new_dp[i + j] = add(new_dp[i + j], mul(dp[i], C[i + j][j]));
        }
        swap(dp, new_dp);
    }
    return dp[k];
}

int get_ans(int len, vector<int> cur_cnt) {
    int ans = 0;
    for (int mx = 0; mx < 10; ++mx) {
        for (int has_mx = cur_cnt[mx]; has_mx <= cur_cnt[mx] + len; ++has_mx) {
            bool ok = true;
            for (int i = 0; i < 10; ++i) {
                if (cur_cnt[i] > has_mx || (cur_cnt[i] == has_mx && i > mx)) {
                    ok = false;
                    break;
                }
            }
            if (ok) {
                vector<int> a(10);
                for (int i = 0; i < mx; ++i)
                    a[i] = has_mx - cur_cnt[i];
                for (int i = mx + 1; i < 10; ++i)
                    a[i] = has_mx  - cur_cnt[i] - 1;
                ans = add(ans, mul(mul(mx, C[len][has_mx - cur_cnt[mx]]), get_cnt(len - (has_mx - cur_cnt[mx]), a)));
            }
        }
    }
    return ans;
}

void solve() {
    string s;

    cin >> s;
    int ans = 0;
    for (int i = 1; i < s.size(); ++i) {
        // up to i are zeroes
        for (int j = 1; j < 10; ++j) {
            vector<int> cur_cnt(10);
            cur_cnt[j] = 1;
            ans = add(ans, get_ans(s.size() - i - 1, cur_cnt));
        }
    }
    vector<int> cur_cnt(10);
    for (int i = 0; i < s.size(); ++i) {
        for (int j = i == 0 ? 1 : 0; j < int(s[i] - '0'); ++j) {
            ++cur_cnt[j];
            ans = add(ans, get_ans(s.size() - i - 1, cur_cnt));
            --cur_cnt[j];
        }
        ++cur_cnt[s[i] - '0'];
    }
    ans = add(ans, get_ans(0, cur_cnt));
    cout << ans << '\n';
}


signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    C[0][0] = 1;
    for (int n = 1; n < MAXN; ++n) {
        C[n][0] = 1;
        for (int k = 1; k <= n; ++k) {
            C[n][k] = add(C[n - 1][k - 1], C[n - 1][k]);
        }
    }

    int t;
    cin >> t;
    while (t--) {
        solve();
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3596kb

input:

5
9
99
999
99999
999999

output:

45
615
6570
597600
5689830

result:

ok 5 number(s): "45 615 6570 597600 5689830"

Test #2:

score: 0
Accepted
time: 1ms
memory: 3648kb

input:

34
7
48
8
76
1
97
7
5
7
7
2
89
9
4
84
46
6
73
86
78
5
3
8
9
31
24
78
7
11
45
2
65
88
6

output:

28
236
36
420
1
597
28
15
28
28
3
525
45
10
484
221
21
399
500
435
15
6
36
45
145
104
435
28
47
215
3
341
516
21

result:

ok 34 numbers

Test #3:

score: 0
Accepted
time: 1ms
memory: 3588kb

input:

16
935
888
429
370
499
881
285
162
178
948
205
858
573
249
773
615

output:

6009
5618
2456
2078
2905
5562
1603
887
993
6121
1174
5378
3333
1374
4724
3631

result:

ok 16 numbers

Test #4:

score: 0
Accepted
time: 1ms
memory: 3660kb

input:

12
1242
9985
6469
9310
4191
9497
3166
3495
9711
9698
4137
2257

output:

7292
63531
37910
58047
23987
59479
18076
19675
61184
61086
23672
12913

result:

ok 12 numbers

Test #5:

score: 0
Accepted
time: 2ms
memory: 3628kb

input:

10
61195
72739
10164
79164
57851
12326
29132
55992
67377
13873

output:

337575
408170
63792
450686
316513
70493
157773
305011
374163
77954

result:

ok 10 numbers

Test #6:

score: 0
Accepted
time: 0ms
memory: 3852kb

input:

8
529983
127270
421121
291729
461233
695056
365028
271160

output:

2744573
687141
2160067
1500426
2359204
3705475
1851172
1381981

result:

ok 8 numbers

Test #7:

score: 0
Accepted
time: 3ms
memory: 3656kb

input:

7
7934351
8474057
1287369
5845624
7796773
5805755
7349121

output:

42465725
45668947
6716401
30094426
41554096
29861098
38756757

result:

ok 7 numbers

Test #8:

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

input:

3
5014252832385738
8762796162648653
919997886706385

output:

892033338
297722019
462512414

result:

ok 3 number(s): "892033338 297722019 462512414"

Test #9:

score: 0
Accepted
time: 37ms
memory: 3820kb

input:

2
775701797726112292362823101
75927988177061355614

output:

371275551
566830847

result:

ok 2 number(s): "371275551 566830847"

Test #10:

score: 0
Accepted
time: 297ms
memory: 3664kb

input:

1
65760982925996012426370962570581226245366145016666

output:

661063035

result:

ok 1 number(s): "661063035"

Test #11:

score: 0
Accepted
time: 300ms
memory: 3652kb

input:

1
62597468169905757754175023836706426691470692832490

output:

9983261

result:

ok 1 number(s): "9983261"

Test #12:

score: 0
Accepted
time: 306ms
memory: 3820kb

input:

1
78912847369504885593964702297317051208901751786824

output:

817123221

result:

ok 1 number(s): "817123221"

Test #13:

score: 0
Accepted
time: 329ms
memory: 3588kb

input:

1
99999999999999999999999999999999999999999999999999

output:

25251932

result:

ok 1 number(s): "25251932"

Test #14:

score: 0
Accepted
time: 220ms
memory: 3648kb

input:

1
999999999999999999999999999999999999999999999

output:

439421821

result:

ok 1 number(s): "439421821"

Test #15:

score: 0
Accepted
time: 142ms
memory: 3652kb

input:

1
9999999999999999999999999999999999999999

output:

387537647

result:

ok 1 number(s): "387537647"

Test #16:

score: 0
Accepted
time: 330ms
memory: 3656kb

input:

1
99999999999999999999999998889999898988888889998888

output:

909431898

result:

ok 1 number(s): "909431898"

Test #17:

score: 0
Accepted
time: 328ms
memory: 3668kb

input:

1
99999999999999999999999998989899988889989889999888

output:

289727470

result:

ok 1 number(s): "289727470"

Test #18:

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

input:

1
99999999999999999999999998998988898888898889898999

output:

962896416

result:

ok 1 number(s): "962896416"

Test #19:

score: 0
Accepted
time: 142ms
memory: 3656kb

input:

1
9999999999999999999989988898888989888899

output:

995903330

result:

ok 1 number(s): "995903330"

Test #20:

score: 0
Accepted
time: 141ms
memory: 3648kb

input:

1
9999999999999999999989999889889998998898

output:

385460258

result:

ok 1 number(s): "385460258"

Test #21:

score: 0
Accepted
time: 329ms
memory: 3632kb

input:

1
99999999999999999999999999999999999999999999999999

output:

25251932

result:

ok 1 number(s): "25251932"