CF1316E Team Building

一、题目

点此看题

二、解法

由于$p$很小，可以考虑状压，设$dp[i][s]$为前$i$个人选出来运动员的状态为$s$，关键是在于如何判断一个人当不当观众。将所有人按$a$排序，所以选前面的人当观众一定比后面的更优，只要观众人数（有的人数$-$运动员数）没满就贪心选。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long
const int M = 100005;
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,k,dp[M][1<<7];
struct node
{
	int x,s[10];
	bool operator < (const node &b) const
	{
		return x>b.x;
	}
}a[M];
signed main()
{
	n=read();m=read();k=read();
	for(int i=1;i<=n;i++)
		a[i].x=read();
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<m;j++)
			a[i].s[j]=read();
	}
	sort(a+1,a+1+n);
	memset(dp,-1,sizeof dp);
	dp[0][0]=0;
	for(int i=1;i<=n;i++)
		for(int j=0;j<(1<<m);j++)
		{
			int t=i-1-__builtin_popcount(j);
			if(t<k && dp[i-1][j]!=-1)
				dp[i][j]=dp[i-1][j]+a[i].x;
			else
				dp[i][j]=dp[i-1][j];
			for(int k=0;k<m;k++)
				if(j&(1<<k) && dp[i-1][j^(1<<k)]!=-1)
					dp[i][j]=max(dp[i][j],dp[i-1][j^(1<<k)]+a[i].s[k]);
		}
	printf("%lld\n",dp[n][(1<<m)-1]);
}