Codeforces 1316E （DP+状态压缩+贪心）

题目链接：Team Building

题意：有n个人,从中选出p个当运动员，k个当观众，每个人当运动员和当观众贡献不同，在运动员中所处位置不同贡献也不同，求如何分配有最大贡献

题解：参考自大佬carcar1，感觉这题挺好的，所以还是写一下记录一下，这两天都忙着蓝桥杯没咋写，，好了进入正文，这题P最大只有7，总能想到状态压缩，再乘上n好像复杂度正好1e7-1e8，我们就可以用dp[i][j]表示前i个人且状态为j时的最优解，但是怎么选观众呢，我们可以先按a降序排，默认初始情况是前k个人（a最大的k个）出观众，然后每次状态转移时看看当前人数i-已经选完运动员的人数是否大于k，即剩余人是否够出观众，如果够，那么当前点更新时的贡献就为这个人担当每个位置时能带来的贡献，否则说明当前点原本是被默认出观众的，那么再让其去当运动员就要减去其当观众的贡献，但是这样就少了一个人出观众，所以还要让下一个既不是观众又不是运动员的出观众（贪心取最大的a,）因为a降序，所以可以求出下一个出观众的人的下标，

细节和不懂见代码及注释：

#include<iostream>
#include<stack>
#include<list>
#include<set>
#include<vector>
#include<algorithm>
#include<math.h>
#include<numeric>
#include<map>
#include<cstring>
#include<queue>
#include<iomanip>
#include<cmath>
#include<queue>
#include <bitset>
#include<unordered_map>/*
	#ifndef local
	#define endl '\n'
#endif */
#define mkp make_pair
using namespace std;
using std::bitset;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll MAXN=2e6+10;
const ll N=1e5+100;
const ll mod=1e9+7;
const ll hash_p1=1610612741;
const ll hash_p2=805306457;
const ll hash_p3=402653189;
//-----------------------------------------------------------------------------------------------------------------*/
// ll head[MAXN],net[MAXN],to[MAXN],edge[MAXN]/*流量*/,cost[MAXN]//费用;
/* 
void add(ll u,ll v,ll w,ll s){
	to[++cnt]=v;net[cnt]=head[u];edge[cnt]=w;cost[cnt]=s;head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];edge[cnt]=0;cost[cnt]=-s;head[v]=cnt;
}
struct elemt{
	int p,v;
};
-----------------------------------
求[1,MAXN]组合式和逆元 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
		b/=2;
	}
	return res;
}
ll fac[MAXN+10],inv[MAXN+10]
ll C(int m,int n){//组合式C(m,n); 
	if(!n){
		return 1;
	}
	return fac[m]*(inv[n]*inv[m-n]%mod)%mod;
}
fac[0]=1;inv[0]=1;
for(ll i=1;i<=MAXN;i++){
	fac[i]=(fac[i-1]*i)%mod;
	inv[i]=mi(fac[i],mod-2);
}
---------------------------------
 unordered_map<int,int>mp;
//优先队列默认小顶堆 ， greater<int> --小顶堆  less<int> --大顶堆  
priority_queue<elemt,vector<elemt>,comp>q;
struct comp{
	public:
		bool operator()(elemt v1,elemt v2){
			return v1.v<v2.v;
		}
};
	set<int>::iterator it=st.begin();
*/
// vector<vector<int>>edge; 二维虚拟储存坐标 
//-----------------------------------------------------------------------------------------------------------------*/
  //map<int,bool>mp[N]; 
ll dp[N][300];//表示前i个点，当前选的运动员状态为j所能产生的最大贡献
struct elemt{//因为要按a降序，所以用结构体存
	ll a;
	ll s[10];
};
elemt d[N];
bool comp(elemt v1,elemt v2){
	return v1.a>v2.a;
}
int cal(int k){//求k中二进制1的个数
	int res=0;
	while(k){
		if(k%2){
			res++;
		}
		k/=2;
	}
	return res;
}
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（不含整数部分）*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（含整数部分）*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
	int n,p,k;
	cin>>n>>p>>k;
	for(int i=1;i<=n;i++){
		cin>>d[i].a;
	}
	for(int i=1;i<=n;i++){
		for(int j=0;j<p;j++){
			cin>>d[i].s[j];
		}
	}
	sort(d+1,d+n+1,comp);//按a降序
	dp[0][0]=0;
	for(int i=1;i<=k;i++){//初始默认前k个出观众
		dp[0][0]+=d[i].a;
	}
	for(int i=1;i<=k;i++){
		dp[i][0]=dp[i-1][0];
	}
	for(int i=1;i<=n;i++){
		for(int j=0;j<(1<<p);j++){//枚举状态
			dp[i][j]=dp[i-1][j];//不替代任何位置
			for(int q=0;q<p;q++){//枚举当前选手去替代第i+1个位置
				if(j&(1<<q)){
					int tmp=cal(j);
					if(i-tmp>=k){//剩下的人足够出观众,
						dp[i][j]=max(dp[i][j],dp[i-1][j^(1<<q)]+d[i].s[q]);
					}
					else{//剩下的人不够出观众，当前又被选中，那么就将尽可能考前的点代替当前出观众
						dp[i][j]=max(dp[i][j],dp[i-1][j^(1<<q)]+d[i].s[q]-d[i].a+d[tmp+k].a);
					}
				}
			}
		}
	}
	cout<<dp[n][(1<<p)-1]<<endl;
	return 0;
}