LOJ2538:「PKUWC2018」Slay the Spire

LOJ

题意

给你$2n$张牌,$n$张攻击牌,$n$张强化牌,每次随机等概率选出$m$张牌,且以最优决策打出$k$张($k\le m$),问期望能造成多少伤害;

题解

因为最后的答案要乘上$\frac{(2n)!}{m!(2n-m)!}$,所以其实要求的就是所有选牌方案以最优决策打出造成的伤害总和;首先这题给了个非常重要的条件保证强化牌的数值大于等于2,这意味着什么呢?意味着如果同时有强化牌和攻击牌时,肯定要优先打出强化牌(当然至少要保证打出一张攻击牌),这个结论是显然的,考虑如果这里有三张牌分别是攻击牌($3$)攻击牌($3$)强化牌($2$),你要打出两张,如果考虑优先打攻击牌那么最优的情况就是现在这种最大攻击力是一样的,那么造成的伤害最多也就是在一张的基础上翻一倍,而强化牌能让那一张牌的攻击力至少翻一倍,这结论就是显然的了;那么现在我们就有哪些打出牌的情况以及怎么抽这样的牌;

• 第一种
  因为要优先考虑强化牌,所以第一种牌型是有强化牌打强化牌,剩下的至少保证打一张攻击牌,同时剩的$m-k$张牌全部抽的是攻击牌;
• 第二种
  上面那种情况肯定是不全面的,为什么剩下的$m-k$张不能抽强化牌呢,现在考虑什么时候剩下的牌会抽到强化牌,那么一定是只打出一张攻击牌时,才能剩余强化牌,否则你打两张打三张攻击牌为什么不先打这剩的强化牌?

所以总结一下,一共只有两种抽牌类型(不打攻击牌的直接不考虑因为伤害为$0$):

1. 抽法:$m$张牌中有$0$到$min(K-1,n)$张强化牌,剩的全是攻击牌
   打法:先把强化牌全部打出,再把剩下的攻击牌按攻击力从大到小打出
2. 抽法:$m$张牌中任意张攻击牌任意张强化牌,加起来为$m$就行了
   打法:只打一张攻击力最高的,剩下的全打强化牌

好了现在思路就非常清晰了,现在就该考虑这么去找这些牌型的牌;
首先考虑第一种,从大到小先把攻击牌和强化牌分别排序,然后定义$F[i][j]$表示到考虑到第$i$张牌时选$j$张牌出来打还要选$m-k$张不打的方案伤害总和,转移为$$F[i][j]=C[n-i][m-k]\ast \sum _{k=0}^{i-1}(F[k][j-1]+A[i])$$$C$为组合数,但是这个这样的话就是$O(n^3)$的复杂度了,绝对跑不过去;但是想一想我们是从哪些地方转移过来的?其实我们是从每一个选$j-1$张牌的方案转移过来的,那么现在修改状态$F[i]$表示按顺序考虑到当前牌时选$i$张打$m-k$张不打的方案伤害总和,再加上一个辅助数组$G[i]$,表示按顺序考虑到当前牌$-1$时选$i$张出来打的方案伤害总和,那么转移是这样的(t表示当前考虑到的牌的序号):$$G[i]=G[i-1]+A[t]\ast C[t-1][i-1]$$$$F[i]=F[i]+(G[i-1]+A[t]\ast C[t-1][i-1])\ast C[n-t][m-k]$$$G$就相当于在原来选$i-1$的基础上每个方案后面跟一个$A[t]$跟的个数就是在前$t-1$张牌中选$i-1$张;
$F$就是在原有的基础上加上当前牌的贡献,因为当前牌是打出的最后一张,所以说要在它后面的$n-t$张牌中来选,乘上选的方案数就好了;
再考虑强化牌怎么选,因为不用考虑多抽,其实就是跟$G$转移差不多的,再前面的基础上添上一张牌就是新的方案贡献,$H[i]$表示考虑到当前这张牌时选$i$张强化的强化值总和:$$H[i]=H[i]+H[i-1]\ast B[t]$$
第一种牌型所需要的东西就准备齐全了,现在怎么算贡献呢,就挨着挨着考虑打多少张攻击牌就完事了:$$An{s}_1=\sum _{i=1}^{k}F[i]\ast H[k-i]$$
好,现在来考虑第二种牌型,需要的东西也比较简单,$W[i]$表示选当前牌作为唯一一张打出去的牌时再多选$i$张不打的方案伤害总和,因为当前牌要打出去,那么其他不打的牌肯定只能选小于等于它的牌$$W[i]=W[i]+A[t]\ast C[n-t][i]$$
然后是强化牌,强化牌肯定要先满足选满$k-1$张那么转移的时候要用一下上面那个$H$数组,$R[i]$表示考虑到当前牌时选$k-1$张打以及$i$张不打的方案数$$W[i]=W[i]+R[k-1]\ast B[t]\ast C[n-t][i]$$
第二部分的贡献也很好算,就是考虑剩下不打的牌中有多少张攻击牌,注意一下不要算$m-k$,因为这部分在第一种牌型中已经算过了:$$An{s}_2=\sum _{i=1}^{m-k-1}W[i]\ast R[m-k-i]$$
把两部分答案加起来就行了,这题就做完了;
(调了半天发现我先读的攻击牌再读的强化牌,吉老师你明明先介绍的攻击牌啊

#include<bits/stdc++.h>
#define Fst first
#define Snd second
#define RG register
#define mp make_pair
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long LL;
typedef long double LD;
typedef unsigned int UI;
typedef unsigned long long ULL;
template<typename T> inline void read(T& x) {
	char c = getchar();
	bool f = false;
	for (x = 0; !isdigit(c); c = getchar()) {
		if (c == '-') {
			f = true;
		}
	}
	for (; isdigit(c); c = getchar()) {
		x = x * 10 + c - '0';
	}
	if (f) {
		x = -x;
	}
}
template<typename T, typename... U> inline void read(T& x, U& ... y) {
	read(x), read(y...);
}
const int MAX=3000,N=MAX+10,P=998244353;
int n,m,K,T;
LL A[N],B[N],C[N][N],F[N],G[N],H[N],W[N],R[N];
bool cmp(int a,int b) {
	return a>b;
}
//#define rua
int main() {
#ifdef rua
	freopen("GG.in","r",stdin);
#endif
	read(T);
	for(int i=0;i<=3000;++i) {
		C[i][0]=1; C[i][i]=1;
		for(int j=1;j<i;++j) {
			C[i][j]=C[i-1][j]+C[i-1][j-1];
			if(C[i][j]>=P) C[i][j]-=P;
		}
	}
	while(T--) {
		mem(F,0); mem(G,0); mem(H,0); mem(W,0); mem(R,0);
		int n,m,K; read(n,m,K);
		for(int i=1;i<=n;++i) read(B[i]);
		for(int i=1;i<=n;++i) read(A[i]);
		sort(A+1,A+n+1,cmp); sort(B+1,B+n+1,cmp);
		H[0]=1;
		for(int i=1;i<=n;++i) {
			for(int j=K;j;--j) {
				(G[j]+=(G[j-1]+1ll*A[i]*C[i-1][j-1]%P)%P)%=P;
				if(n-i>=m-K) (F[j]+=1ll*(G[j-1]+1ll*A[i]*C[i-1][j-1]%P)%P*C[n-i][m-K]%P)%=P;
				if(j==1) {
					for(int k=0;k<=m-K-1; ++k) (W[k]+=1ll*A[i]*C[n-i][k])%=P;
				}
				(H[j]+=(1ll*H[j-1]*B[i]%P))%=P;
				if(j==K-1) {
					for(int k=1;k<=m-K;++k) (R[k]+=1ll*H[j-1]*B[i]%P*C[n-i][k])%=P;
				}
			}
		}
		if(K==1) {
			for(int i=1;i<=m-K;++i) R[i]=C[n][i];
		}
		int res=0;
		for(int i=0;i<=m-K-1;++i) {
			int t=m-K-i;
			(res+=1ll*W[i]*R[t]%P)%=P;
		}
		for(int i=1;i<=K;++i) (res+=1ll*F[i]*H[K-i]%P)%=P;
		printf("%d\n",res);
	}	
	return 0;
}