QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#234374 | #6522. Digit Mode | ucup-team1198# | AC ✓ | 330ms | 3852kb | C++20 | 2.9kb | 2023-11-01 16:42:42 | 2023-11-01 16:42:42 |
Judging History
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"