QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#139587 | #6522. Digit Mode | ethening | AC ✓ | 460ms | 3584kb | C++20 | 2.1kb | 2023-08-13 23:10:01 | 2023-08-13 23:10:03 |
Judging History
answer
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
#include "bits/stdc++.h"
using namespace std;
using ll = long long;
using pii = pair<int, int>;
ll MOD = 1000000007ll;
ll fac[51], inv[51], ifac[51];
void pre_work() {
fac[0] = inv[0] = ifac[0] = 1;
fac[1] = inv[1] = ifac[1] = 1;
for (int i = 2; i <= 50; i++) {
fac[i] = fac[i - 1] * i % MOD;
inv[i] = inv[MOD % i] * (MOD - MOD / i) % MOD;
ifac[i] = ifac[i - 1] * inv[i] % MOD;
}
}
ll f(int n, int fixed, vector<int>& pref) {
int unfixed = n - fixed;
int mx_pref = *max_element(begin(pref), end(pref));
ll ans = 0;
for (int mode = 1; mode <= 9; mode++) {
for (int freq = mx_pref; freq <= unfixed + pref[mode]; freq++) {
int mxj = unfixed - (freq - pref[mode]);
auto knapsack = [&]() {
vector<ll> dp(mxj + 1, 0);
dp[0] = 1;
for (int i = 0; i <= 9; i++) {
if (i == mode) continue;
int mxk = freq - (i > mode ? 1 : 0) - pref[i];
if (mxk < 0) return 0ll;
for (int j = mxj; j >= 1; j--) {
for (int k = 1; k <= mxk && k <= j; k++) {
dp[j] = (dp[j] + dp[j - k] * ifac[k]) % MOD;
}
}
}
return dp[mxj];
};
ans = (ans + knapsack() * fac[unfixed] % MOD * ifac[freq - pref[mode]] % MOD * mode % MOD);
}
}
return ans;
}
void solve() {
string s;
cin >> s;
int n = s.size();
// cout << "cur: " << s << endl;
ll ans = 0;
vector<int> freq(10);
for (int i = 1; i <= n - 1; i++) {
for (int j = 1; j <= 9; j++) {
freq[j] = 1;
ll req = f(i, 1, freq);
ans = (ans + req) % MOD;
freq[j] = 0;
}
}
for (int i = 0; i < n; i++) {
int dig = s[i] - '0';
for (int j = 0; j < dig; j++) {
if (i == 0 && j == 0) continue;
// cout << "#" << i << " " << j << endl;
freq[j]++;
ll req = f(n, i + 1, freq);
ans = (ans + req) % MOD;
freq[j]--;
}
freq[dig]++;
}
ans += f(n, n, freq);
ans %= MOD;
cout << ans << "\n";
}
int main() {
pre_work();
cin.tie(0)->sync_with_stdio(0);
int t;
cin >> t;
while (t--) {
solve();
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3516kb
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: 3460kb
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: 0ms
memory: 3564kb
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: 3580kb
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: 3456kb
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: 2ms
memory: 3492kb
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: 2ms
memory: 3536kb
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: 12ms
memory: 3584kb
input:
3 5014252832385738 8762796162648653 919997886706385
output:
892033338 297722019 462512414
result:
ok 3 number(s): "892033338 297722019 462512414"
Test #9:
score: 0
Accepted
time: 39ms
memory: 3520kb
input:
2 775701797726112292362823101 75927988177061355614
output:
371275551 566830847
result:
ok 2 number(s): "371275551 566830847"
Test #10:
score: 0
Accepted
time: 396ms
memory: 3456kb
input:
1 65760982925996012426370962570581226245366145016666
output:
661063035
result:
ok 1 number(s): "661063035"
Test #11:
score: 0
Accepted
time: 409ms
memory: 3532kb
input:
1 62597468169905757754175023836706426691470692832490
output:
9983261
result:
ok 1 number(s): "9983261"
Test #12:
score: 0
Accepted
time: 410ms
memory: 3468kb
input:
1 78912847369504885593964702297317051208901751786824
output:
817123221
result:
ok 1 number(s): "817123221"
Test #13:
score: 0
Accepted
time: 455ms
memory: 3524kb
input:
1 99999999999999999999999999999999999999999999999999
output:
25251932
result:
ok 1 number(s): "25251932"
Test #14:
score: 0
Accepted
time: 295ms
memory: 3500kb
input:
1 999999999999999999999999999999999999999999999
output:
439421821
result:
ok 1 number(s): "439421821"
Test #15:
score: 0
Accepted
time: 186ms
memory: 3460kb
input:
1 9999999999999999999999999999999999999999
output:
387537647
result:
ok 1 number(s): "387537647"
Test #16:
score: 0
Accepted
time: 460ms
memory: 3524kb
input:
1 99999999999999999999999998889999898988888889998888
output:
909431898
result:
ok 1 number(s): "909431898"
Test #17:
score: 0
Accepted
time: 459ms
memory: 3464kb
input:
1 99999999999999999999999998989899988889989889999888
output:
289727470
result:
ok 1 number(s): "289727470"
Test #18:
score: 0
Accepted
time: 459ms
memory: 3484kb
input:
1 99999999999999999999999998998988898888898889898999
output:
962896416
result:
ok 1 number(s): "962896416"
Test #19:
score: 0
Accepted
time: 186ms
memory: 3472kb
input:
1 9999999999999999999989988898888989888899
output:
995903330
result:
ok 1 number(s): "995903330"
Test #20:
score: 0
Accepted
time: 186ms
memory: 3452kb
input:
1 9999999999999999999989999889889998998898
output:
385460258
result:
ok 1 number(s): "385460258"
Test #21:
score: 0
Accepted
time: 460ms
memory: 3520kb
input:
1 99999999999999999999999999999999999999999999999999
output:
25251932
result:
ok 1 number(s): "25251932"