Codeforces1316 E. Team Building（状压dp，排序）

题意：

你要组件一支排球队。
有n个人，你需要选出p个人，分别站在p个位置上。
另外还需要从剩下的人中选出k个人作为观众。
对于第i个人，如果作为观众能够为队伍增加a(i)点力量，
如果作为球队队员，在第j个位置上能增加s(i,j)点力量。

问这支排球队的最大力量和是多少。

数据范围：n<=1e6，1<=p<=7，1<=k，k+p<=n

解法：

p很小,容易想到状压dp,
如果没有观众的话,这个题就是状压dp的水题,难点就在于如何处理观众.

将所有人按照a[i]从大到小排序.
令d[i][s]表示前i个人,队员选择状态为s的最大值,s的第j位为1表示第j个位置有人(下标为[0,p-1])
对于枚举到的第i个人:
1.考虑这个人不作为队员:
设目前状态s的队员总数为cnt,如果i<=cnt+k,那么说明观众没满,
因为a[i]是从大到小排序的,如果当前人不作为队员,那么一定要作为观众.
d[i][s]=max(d[i][s],d[i-1][s]+a[i])
2.考虑这个人作为队员:
那么当前s中有一个位置是这个队员的,枚举这个位置,
d[i][s]=max(d[i][s],d[i-1][s^(1<<j)]+a[i][j]),其中s的第j位为1.

需要注意的点是dp数组一开始要初始化为-inf

参考:
https://www.luogu.com.cn/blog/syksykCCC/solution-cf1316e

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e5+5;
struct Node{
    int v;
    int s[8];
}a[maxm];
int d[maxm][1<<8];
int n,p,k;
bool cmp(Node a,Node b){
    return a.v>b.v;
}
int cal(int x){
    int ans=0;
    while(x){
        ans++;
        x&=(x-1);
    }
    return ans;
}
signed main(){
    ios::sync_with_stdio(0);
    cin>>n>>p>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i].v;
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<p;j++){
            cin>>a[i].s[j];
        }
    }
    sort(a+1,a+1+n,cmp);
    for(int i=0;i<=n;i++){
        for(int j=0;j<(1<<p);j++){
            d[i][j]=-1e18;
        }
    }
    d[0][0]=0;
    for(int i=1;i<=n;i++){
        for(int s=0;s<(1<<p);s++){
            d[i][s]=d[i-1][s];
            int cnt=cal(s);
            //考虑不作为队员,作为观众:
            if(i<=cnt+k){//如果没选满k个观众,那么一定作为观众
                d[i][s]=d[i-1][s]+a[i].v;
            }
            //考虑作为队员:
            for(int j=0;j<p;j++){
                if(s>>j&1){
                    d[i][s]=max(d[i][s],d[i-1][s^(1<<j)]+a[i].s[j]);
                }
            }
        }
    }
    cout<<d[n][(1<<p)-1]<<endl;
    return 0;
}

---