bzoj1334 & 洛谷P4394 选举 01背包

题目链接：传送门

题目描述
$N$个政党要组成一个联合内阁，每个党都有自己的席位数.

现在希望你找出一种方案，你选中的党的席位数要大于总数的一半，并且联合内阁的席位数越多越好.

对于一个联合内阁，如果某个政党退出后，其它党的席位仍大于总数的一半，则这个政党被称为是多余的，这是不允许的.

输入格式
第一行给出有多少个政党.其值小于等于300

下面给出每个政党的席位数.总席位数小于等于 100000

输出格式
你的组阁方案中最多能占多少个席位.

观察题面，每个政党只能选一次，想到使用01背包，把席位数当作背包的容量。
但题目又要求总席位数超过所有席位的一半，去除最少席位的政党后总席位数不超过总数的一半。
如何保证这一点呢？
考虑把所有政党按照席位数从大到小排序，然后每次跑01背包时，假设当前枚举到第$i$个政党。

背包的容量上界应设为$a[i]+sum$，其中$a[i]$表示第$i$个政党的席位数，$sum$表示总席位数的一半。

这样可以保证把背包内每个政党删去，剩下的席位数都不超过$sum$，满足题意。
其余部分按照普通01背包处理即可。

代码：

#include<stdio.h>
#include<cstring>
#include<algorithm>
#define re register int
using namespace std;
typedef long long ll;
int read() {
	re x=0,f=1;
	char ch=getchar();
	while(ch<'0' || ch>'9') {
		if(ch=='-')	f=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9') {
		x=10*x+ch-'0';
		ch=getchar();
	}
	return x*f;
}
const int Size=200005;
int n,a[Size];
bool dp[Size];	//dp[i]:记录选出的政党席位数之和为i时是否可行 
inline bool comp(int x,int y) {
	return x>y;
}
int main() {
	n=read();
	int sum=0,ans=0;
	for(re i=1; i<=n; i++) {
		a[i]=read();
		sum+=a[i];
	}
	sum>>=1;
	//把每个政党的席位数从大到小排序 
	sort(a+1,a+1+n,comp);
	//所有政党都不选，席位数之和为0，可行 
	//只选第一个政党可行 
	dp[0]=dp[a[1]]=true;
	for(re i=2; i<=n; i++) {
		//应保证删去每个政党后总席位数不超过总数的一半 
		for(re j=a[i]+sum; j>=a[i]; j--) {
			if(dp[j-a[i]]) {
				dp[j]=true;
				//当前席位数之和比ans大，更新答案 
				if(j>ans)	ans=j;
			}
		}
	}
	printf("%d",ans);
	return 0;
}