QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#883733 | #6410. Classical DP Problem | OIer_wst | WA | 29ms | 199780kb | C++14 | 1.1kb | 2025-02-05 18:32:40 | 2025-02-05 18:32:41 |
Judging History
answer
#include <bits/stdc++.h>
using lint = long long;
const int N = 5e3 + 10;
const int MOD = 998244353;
lint cnt, f[N][N];
int n, k, a[N], b[N];
void input() {
std::cin >> n;
for (int i = 1; i <= n; ++i)
std::cin >> a[i];
std::reverse(a + 1, a + n + 1);
}
int get_ans() {
int p = 0;
while (p + 1 <= n && p + 1 <= a[p + 1]) ++p;
return p;
}
int dp(int m) { // m:剩余的关键列
memset(f, 0, sizeof(f));
f[0][0] = 1;
for (int i = 0; i < k; ++i) {
for (int j = 0; j <= m; ++j) {
(f[i + 1][j + 1] += f[i][j] * (m - j) % MOD) %= MOD;
(f[i + 1][j] += f[i][j] * (a[i + 1] - m + j) % MOD) %= MOD;
}
}
return f[k][m];
}
void T() { // 矩阵转置
for (int i = 1; i <= n; ++i) ++b[a[i]];
for (int i = n - 1; i; --i) b[i] += b[i + 1];
memcpy(a, b, sizeof(b));
}
int fact() {
lint res = 1;
for (int i = 2; i <= k; ++i) (res *= i) %= MOD;
return res;
}
int main() {
input();
k = get_ans();
(cnt += dp(a[k + 1])) %= MOD;
T();
(cnt += dp(a[k + 1])) %= MOD;
std::cout << k << " " << (cnt - fact()) << std::endl;
return 0;
}
詳細信息
Test #1:
score: 100
Accepted
time: 26ms
memory: 199732kb
input:
3 1 2 3
output:
2 6
result:
ok 2 number(s): "2 6"
Test #2:
score: 0
Accepted
time: 24ms
memory: 199612kb
input:
1 1
output:
1 1
result:
ok 2 number(s): "1 1"
Test #3:
score: 0
Accepted
time: 21ms
memory: 199716kb
input:
2 1 1
output:
1 2
result:
ok 2 number(s): "1 2"
Test #4:
score: 0
Accepted
time: 24ms
memory: 199736kb
input:
2 2 2
output:
2 6
result:
ok 2 number(s): "2 6"
Test #5:
score: 0
Accepted
time: 29ms
memory: 199680kb
input:
3 1 1 1
output:
1 3
result:
ok 2 number(s): "1 3"
Test #6:
score: 0
Accepted
time: 24ms
memory: 199720kb
input:
3 2 2 2
output:
2 9
result:
ok 2 number(s): "2 9"
Test #7:
score: 0
Accepted
time: 23ms
memory: 199776kb
input:
3 3 3 3
output:
3 48
result:
ok 2 number(s): "3 48"
Test #8:
score: 0
Accepted
time: 27ms
memory: 199592kb
input:
5 1 1 3 3 4
output:
3 47
result:
ok 2 number(s): "3 47"
Test #9:
score: 0
Accepted
time: 27ms
memory: 199780kb
input:
10 2 4 5 5 5 5 6 8 8 10
output:
5 864
result:
ok 2 number(s): "5 864"
Test #10:
score: -100
Wrong Answer
time: 26ms
memory: 199736kb
input:
30 6 8 9 9 9 10 13 14 15 15 16 17 17 18 20 22 22 23 23 24 24 25 25 25 27 28 28 29 29 30
output:
17 -12054489
result:
wrong answer 2nd numbers differ - expected: '986189864', found: '-12054489'