题意:
两个人A和B分别有n和m张卡牌,每张牌上有一个数字。
两人需要分别选出k张牌,然后将他们按数字排序后一 一配对比较大小。
问有多少种方案,使得A的k张卡牌上的数字都比B对应的牌要大
数据范围:n,m<=1e3,k<=10
解法:
这题n,m<=1e3,可以O(n^2*k),
但是claris讲的题是n,m<=3e4的,用的O(n*k^2)
以下A的牌我简称A,同理B的牌我简称B.
将A和B放到一起排序,如果A和B的值一样则优先把A放到左边
因为题目要求的是严格大于,所以每个A左边的必须是严格小于它的
然后问题变成,从这个序列中选出2k个数,其中k个是A,k个是B,且序列合法
如何判断序列是否合法呢?
将B看成左括号,A看成右括号,那么需要满足括号匹配的合法性(前缀和>=0)
d[i][x][y]表示前i个数,x个B,y个A的合法方案数
总复杂度O(n*k^2)
code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=2e3+5;
const int mod=1e9+9;
struct Node{
int v;
bool isB;
bool operator<(const Node& a)const{
if(v==a.v)return !isB;//AB相等的时候优先把A放在左边,因为A要严格大于B
return v<a.v;//从小到大排序
}
}a[maxm];
int d[maxm][12][12];
int n,m,k;
signed main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++){
scanf("%d",&a[i].v);
a[i].isB=0;
}
for(int i=1;i<=m;i++){
scanf("%d",&a[i+n].v);
a[i+n].isB=1;
}
sort(a+1,a+1+n+m);
d[0][0][0]=1;
for(int i=1;i<=n+m;i++){
for(int x=0;x<=k;x++){
for(int y=0;y<=x;y++){
d[i][x][y]=d[i-1][x][y];
if(a[i].isB){//B,左括号
if(x)d[i][x][y]=(d[i][x][y]+d[i-1][x-1][y])%mod;
}else{//A,右括号
if(y)d[i][x][y]=(d[i][x][y]+d[i-1][x][y-1])%mod;
}
}
}
}
printf("%d\n",d[n+m][k][k]);
return 0;
}