QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#615188 | #9436. Some Sum of Subset | ucup-team5075# | WA | 175ms | 192440kb | C++14 | 1.8kb | 2024-10-05 17:48:44 | 2024-10-05 17:48:45 |
Judging History
answer
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define int long long
const int N=3009,mod=998244353;
int dp[2][N][N],ans[N];
int sum[N][N];
int C[N][N],Csum[N][N],cf[N];
int cnt[N],cnts[N];
int n,m,a[N];
void Add(int &x,int y){if(y<0) exit(-1);x+=y;if(x>=mod) x-=mod;}
signed main()
{
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]),cnt[a[i]]++;
// n=3000;m=1;
// for(int i=1;i<=n;i++) a[i]=rand()%3000+1,cnt[a[i]]++;
for(int i=0;i<=3000;i++)
{
C[i][0]=Csum[i][0]=1;
for(int j=1;j<=i;j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
for(int j=1;j<=3000;j++) Csum[i][j]=(Csum[i][j-1]+C[i][j])%mod;
}
cf[0]=1;
for(int i=1;i<=3000;i++) cf[i]=cf[i-1]*2%mod;
for(int i=1;i<=3000;i++) cnts[i]=cnts[i-1]+cnt[i];
dp[1][0][0]=1;
for(int i=3000;i>=1;i--)
{
for(int j=0;(i+1)*j<=m;j++)
for(int k=m-1;k>=0;k--)
sum[j][k]=(sum[j][k+1]+dp[(i+1)&1][j][k])%mod;
if(i<m)
{
for(int j=0;(i+1)*j<m;j++)
for(int l=1;i*l<=m&&l<=cnt[i];l++)
{
for(int d=l;d<=cnt[i];d++)
for(int kk=0;kk<=cnts[i]-d;kk++)
Add(ans[kk],(sum[j][m-i*l]-sum[j][m-i*(l-1)]+mod)%mod
*C[d-1][l-1]%mod
*(Csum[cnts[i]-d][cnts[i]-d]-Csum[cnts[i]-d][kk]+C[cnts[i]-d][kk]+mod)%mod);
}
}
// cerr<<i<<endl;
for(int j=0;i*j<m;j++)
for(int k=0;k<m;k++)
for(int l=0;l<=j&&l<=cnt[i]&&l*i<=k;l++)
Add(dp[i&1][j][k],dp[(i+1)&1][j-l][k-l*i]);
for(int j=0;(i+1)*j<m;j++)
for(int k=0;k<m;k++)
dp[(i+1)&1][j][k]=0;
// cerr<<i<<" : \n";
// for(int i=0;i<=n;i++) printf("%lld ",ans[i]);
// puts("");
}
for(int i=0;i<=n;i++) Add(ans[i],((Csum[n][n]-Csum[n][i])-(Csum[cnts[m-1]][n]-Csum[cnts[m-1]][i])+mod*3)%mod);
for(int i=0;i<=n;i++) printf("%lld\n",ans[i]);
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 39ms
memory: 130496kb
input:
4 7 3 1 5 2
output:
6 4 1 0 0
result:
ok 5 tokens
Test #2:
score: 0
Accepted
time: 35ms
memory: 130808kb
input:
1 5 7
output:
1 0
result:
ok 2 tokens
Test #3:
score: 0
Accepted
time: 45ms
memory: 130832kb
input:
9 18 1 9 5 6 2 7 1 4 8
output:
346 309 230 126 46 10 1 0 0 0
result:
ok 10 tokens
Test #4:
score: 0
Accepted
time: 138ms
memory: 175368kb
input:
1 1467 556
output:
0 0
result:
ok 2 tokens
Test #5:
score: 0
Accepted
time: 175ms
memory: 191092kb
input:
1 1753 2514
output:
1 0
result:
ok 2 tokens
Test #6:
score: 0
Accepted
time: 111ms
memory: 162500kb
input:
1 1182 1182
output:
1 0
result:
ok 2 tokens
Test #7:
score: 0
Accepted
time: 168ms
memory: 192440kb
input:
2 1741 955 835
output:
1 0 0
result:
ok 3 tokens
Test #8:
score: 0
Accepted
time: 131ms
memory: 177116kb
input:
2 1481 2004 1570
output:
3 1 0
result:
ok 3 tokens
Test #9:
score: 0
Accepted
time: 108ms
memory: 170192kb
input:
2 1336 1336 1336
output:
3 1 0
result:
ok 3 tokens
Test #10:
score: 0
Accepted
time: 54ms
memory: 137300kb
input:
12 400 2163 2729 1322 2787 2404 1068 1502 746 898 2057 1771 502
output:
4095 4083 4017 3797 3302 2510 1586 794 299 79 13 1 0
result:
ok 13 tokens
Test #11:
score: 0
Accepted
time: 165ms
memory: 183584kb
input:
42 1609 532 722 2650 2889 2260 659 1831 401 2779 1262 2185 1479 515 552 1627 637 1080 580 1589 2154 2650 219 2924 1712 311 2609 1062 968 1357 1310 1942 2419 2465 300 2546 2537 2967 1197 2271 1551 999 2531
output:
780135870 780135476 780133126 780115746 779984839 779086101 773744618 746617208 628409426 182368401 707355349 419830171 342550392 778349940 825220795 979080057 177969332 68371210 748001250 187917550 492152134 287983982 592218358 32133349 711757614 602140473 799225822 952989153 1466429 437088327 3596...
result:
ok 43 tokens
Test #12:
score: 0
Accepted
time: 48ms
memory: 130460kb
input:
93 24 1687 2324 1216 2696 2884 1678 1301 1957 1247 526 2519 2252 2141 960 1505 175 315 388 440 892 837 1942 256 1667 641 1629 2565 2837 1200 2427 418 1072 427 1589 969 886 2991 2249 539 1132 949 2425 2616 2107 1259 582 1979 386 177 1860 2825 316 2004 956 1218 2332 1498 2939 1126 1706 404 1604 505 11...
output:
958346177 958346084 958341806 958212040 955292305 903321022 141075538 649896624 629379357 102857051 471642575 350313411 852223261 984774652 743966819 740155079 222450670 637406264 282036249 875748580 177575581 602719974 451360351 244531346 556337405 139169356 737451056 595307440 402860967 935337558 ...
result:
ok 94 tokens
Test #13:
score: -100
Wrong Answer
time: 99ms
memory: 154472kb
input:
162 1057 695 2739 2637 797 503 1051 2243 2429 160 1900 1798 1584 1488 1377 2764 1324 978 2696 913 2651 817 1522 819 314 882 1536 2785 2193 540 2417 1357 959 2765 408 1693 1305 1643 2565 2677 966 860 2547 936 2123 496 2098 2130 38 378 1395 2725 2657 2107 2096 2039 2421 2548 431 2205 2907 1085 2919 25...
output:
120975728 120972094 120923625 119819665 87996541 176739802 16143169 43761708 181003016 894357661 749430532 209621212 798287877 119735182 249473423 493865870 691723073 381797736 542826224 63865434 466502295 297112756 115675471 686893187 226591690 291149805 272381235 220037420 897019520 543603827 1308...
result:
wrong answer 1st words differ - expected: '389411124', found: '120975728'