QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#282430 | #6410. Classical DP Problem | Giulian617 | WA | 16ms | 103500kb | C++23 | 1.5kb | 2023-12-12 00:31:42 | 2023-12-12 00:31:43 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
const int NMAX=5005,MOD=998244353;
int n,r,a[NMAX],b[NMAX],aux[NMAX][NMAX],dp[NMAX][NMAX];
///dp[i][j] = number of ways to cover the first i lines and j columns
int find_r()
{
int ans=1;
while(ans<=n && a[ans]>=ans)
ans++;
return ans-1;
}
void compute_transpose()
{
for(int i=1; i<=n; i++)
for(int j=1; j<=a[i]; j++)
aux[i][j]=1;
for(int j=1; j<=n; j++)
for(int i=1; i<=n; i++)
b[j]+=aux[i][j];
}
int solve(int *a)
{
int x=a[r+1];///number of lines/columns we must occupy
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(int i=0; i<r; i++)
for(int j=0; j<r; j++)
{
if(j<x)///we still have to occupy some lines/columns
dp[i+1][j+1]=(dp[i+1][j+1]+(1LL*(x-j)*dp[i][j])%MOD)%MOD;
dp[i+1][j]=(dp[i+1][j]+(1LL*(a[i+1]-(x-j))*dp[i][j])%MOD)%MOD;
}
return dp[r][x];
}
int find_w(int r)
{
int fact=1;
compute_transpose();
for(int i=1; i<=r; i++)
fact=(fact*i)%MOD;
return ((solve(a)+solve(b))%MOD-fact+MOD)%MOD;///ansLines+ansColumns-factorial
///if we don't need to compute an answer for the lines
///or for the columns, we will get factorial from those functions
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin>>n;
for(int i=1; i<=n; i++)
cin>>a[n-i+1];
r=find_r();
cout<<r<<' '<<find_w(r);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 11ms
memory: 103464kb
input:
3 1 2 3
output:
2 6
result:
ok 2 number(s): "2 6"
Test #2:
score: 0
Accepted
time: 4ms
memory: 101628kb
input:
1 1
output:
1 1
result:
ok 2 number(s): "1 1"
Test #3:
score: 0
Accepted
time: 16ms
memory: 103500kb
input:
2 1 1
output:
1 2
result:
ok 2 number(s): "1 2"
Test #4:
score: 0
Accepted
time: 8ms
memory: 101756kb
input:
2 2 2
output:
2 6
result:
ok 2 number(s): "2 6"
Test #5:
score: 0
Accepted
time: 11ms
memory: 103316kb
input:
3 1 1 1
output:
1 3
result:
ok 2 number(s): "1 3"
Test #6:
score: 0
Accepted
time: 12ms
memory: 102724kb
input:
3 2 2 2
output:
2 9
result:
ok 2 number(s): "2 9"
Test #7:
score: 0
Accepted
time: 7ms
memory: 102360kb
input:
3 3 3 3
output:
3 48
result:
ok 2 number(s): "3 48"
Test #8:
score: 0
Accepted
time: 8ms
memory: 102436kb
input:
5 1 1 3 3 4
output:
3 47
result:
ok 2 number(s): "3 47"
Test #9:
score: 0
Accepted
time: 12ms
memory: 101988kb
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: 7ms
memory: 102480kb
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 843284081
result:
wrong answer 2nd numbers differ - expected: '986189864', found: '843284081'