QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#768730 | #7875. Queue Sorting | xingrin | WA | 35ms | 9024kb | C++14 | 2.6kb | 2024-11-21 14:05:15 | 2024-11-21 14:05:15 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
// using ll=long double;
using pii=pair<ll,ll>;
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
const int N = 510, mod = 998244353;
ll f[N][N];
ll a[N];
ll len[N];
ll c[N][N];
ll sumc[N][N];
void init()
{
for(int i=0;i<=500;i++)
{
for(int j=0;j<=i;j++)
{
if(!j) c[i][j] = 1;
else c[i][j] = (c[i-1][j-1] + c[i-1][j])%mod;
}
}
for(int i=0;i<=500;i++)
{
for(int j=0;j<=i;j++)
{
if(!j) sumc[i][j] = c[i][j];
else sumc[i][j] = (sumc[i][j-1] + c[i][j])%mod;
}
}
}
void solve()
{
ll n;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i], len[i] = len[i-1] + a[i];
f[0][0] = 1;
f[1][0] = 1;
for(int i=2;i<=n;i++)
{
for(int j =0;j<len[i-1];j++)
{
f[i][j] += f[i-1][j];
if(a[i] == 0) continue;
for(int k=j+1; k<len[i]; k++)
{
if(k-j<=a[i] && k-j-1 <= len[i-1]-j-1)
{
f[i][k] += f[i-1][j] * sumc[k-j-1][k-j-1]%mod;
f[i][k]%=mod;
}
else if(k-j>a[i] && k-j-1<=len[i-1]-j-1)
{
// cout<<11111<<endl;
ll dlt = (k-j) - a[i];
f[i][k] += f[i-1][j] * (sumc[k-j-1][k-j-1] - sumc[k-j-1][dlt-1])%mod;
f[i][k]%=mod;
f[i][k] = (f[i][k] + mod)%mod;
}
else if(k-j<=a[i] && k-j-1>len[i-1]-j-1)
{
ll dlt = (k-j-1) - (len[i-1] - j-1);
f[i][k] += f[i-1][j] * sumc[k-j-1][k-j-1-dlt]%mod;
f[i][k]%=mod;
}
else
{
ll dlt1 = (k-j) - a[i];
ll dlt2 = (k-j-1) - (len[i-1] - j-1);
if(dlt1 >= k-j-dlt2) continue;
f[i][k] += f[i-1][j] * (sumc[k-j-1][k-j-1-dlt2] - sumc[k-j-1][dlt1-1])%mod;
f[i][k]%=mod;
f[i][k] = (f[i][k] + mod)%mod;
}
}
}
}
ll ans = 0;
// cout<<f[2][0]<<' '<<f[2][1]<<' '<<f[2][2]<<' '<<f[2][3]<<endl;
for(int i=0;i<=500;i++)
{
ans = (ans + f[n][i])%mod;
}
cout<<ans;
}
int main(){
IOS;
int t=1;
init();
// cin>>t;
while(t--) solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 2ms
memory: 9024kb
input:
4 1 1 1 1
output:
14
result:
ok 1 number(s): "14"
Test #2:
score: -100
Wrong Answer
time: 35ms
memory: 8980kb
input:
300 0 5 2 2 1 0 3 2 2 5 2 1 1 2 1 3 2 3 2 0 0 0 0 1 2 2 3 0 2 2 3 2 0 2 3 0 6 0 0 2 0 1 3 2 1 1 1 3 4 0 1 0 4 1 1 1 1 1 1 2 3 2 1 2 3 2 3 0 5 3 3 2 0 1 1 0 2 1 1 2 0 0 2 1 1 3 2 2 1 2 1 3 0 3 0 1 2 2 0 5 0 2 2 0 0 0 1 2 1 4 2 1 1 0 3 0 2 0 3 1 1 2 0 2 1 1 0 2 0 1 2 2 3 3 1 1 1 1 0 1 3 3 1 0 2 2 4 2 ...
output:
0
result:
wrong answer 1st numbers differ - expected: '507010274', found: '0'