bzoj4742 [Usaco2016 Dec]Team Building

题目

一眼dp题，对吧，然后就是考虑状态的表示与转移了。

有两个数组，f[i][j]是肯定要有的，然后选k个，最终状态就为f[i][j][k]就好了。表示a中前i个与b中前j个，选出了k个的方案数。（当然顺序对于这道题没有影响，但是要先排一个序。）

转移：

$f[i][j][k]=f[i-1][j][k]+f[i][j-1][k]-f[i-1][j-1][k]$
这是不考虑新加方案的式子，还是比较好像的，就像矩阵求和一样。

如果a[i]>b[j]的话，就会有转移：
$f[i][j][k]=f[i][j][k]+f[i-1][j-1][k-1]$

之后就是一些边界处理了。

#include<bits/stdc++.h>
#define mod 1000000009
using namespace std;
int n,m,K;
int a[1005],b[1005];
int f[1005][1005][15];
int main()
{
    freopen("in.txt","r",stdin);
    scanf("%d%d%d",&n,&m,&K);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)scanf("%d",&b[i]);
    sort(a+1,a+n+1),sort(b+1,b+m+1);
    f[0][0][0]=1;
    for(int i=1;i<=n;i++)f[i][0][0]=1;
    for(int i=1;i<=m;i++)f[0][i][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            f[i][j][0]=1;
            for(int k=1;k<=K;k++)
            {
                if(k>i||k>j)break;
                f[i][j][k]=(f[i][j][k]+f[i][j-1][k])%mod;
                f[i][j][k]=(f[i][j][k]+f[i-1][j][k])%mod;
                f[i][j][k]=(f[i][j][k]-f[i-1][j-1][k]+mod)%mod;
                if(a[i]>b[j])f[i][j][k]=(f[i][j][k]+f[i-1][j-1][k-1])%mod;
            }
        }
    cout<<f[n][m][K];
    return 0;
}