题目
https://atcoder.jp/contests/agc054/tasks/agc054_b
思路
怎么看,这题应该要DP,但是描述一个状态是很大问题:你怎么知道选了哪些橙子(共有(2^{100})种情况)?
经过一番漫长的思考后,师父,我悟了!
其实,并不用描述选了哪些橙子,题目的关键就在排列(P)和那两人拿橘子的顺序是一一对应的,比如说,Takahashi 依次取了1,5,2号橙子,Aoki依次取了4,3号橙子,那么,有且仅有一个(P)和这种情况对应,且该(P)只对应这种情况(用心体会下)
这样,问题就简单多了,设(f_i)表示拿了(i)个橙子,且它们的重量和恰好等于所有橙子重量和的一半,的方案数.答案就应该是(别忘了取模):
[ans=sum^n_{i=1}f_icdot i!cdot (n-i)!
]
(i!)是其中一个人拿橙子的顺序的方案数,((n-i)!)就是另一个人拿橙子的顺序的方案数
如何得到(f)?
DP:(f_{i,j})表示拿了(i)个橙子,总重量为(j)的方案数,过程有点像背包问题,应该都会了吧
代码
#include <iostream>
#include <cstdio>
using namespace std;
#define N 110
#define ll long long
#define mod 998244353ll
ll f[N][10010];
ll n , w[N] , sumw;
ll fac[N];
int main() {
fac[0] = 1;
for(int i = 1 ; i <= 100 ; i++)
fac[i] = fac[i - 1] * i % mod;
cin >> n;
for(int i = 1 ; i <= n ; i++) {
cin >> w[i];
sumw += w[i];
}
if(sumw % 2 == 1) {
cout << 0;
return 0;
}
f[0][0] = 1;
for(int i = 1 ; i <= n ; i++)//枚举第i个橙子
for(int j = n ; j >= 0 ; j--)//一共j个橙子(除了i),注意倒叙枚举
for(int k = sumw ; k >= 0 ; k--)//重量(除了i)
f[j + 1][k + w[i]] += f[j][k],
f[j + 1][k + w[i]] %= mod;
ll ans = 0;
for(int i = 1 ; i <= n ; i++)//求答案
ans = (ans + fac[i] * fac[n - i] % mod * f[i][sumw / 2] % mod) % mod;
cout << ans << endl;
return 0;
}