【Atcoder agc020 C】Median Sum,序列子集和中位数,bitset,01背包

problem

C - Median Sum / Time Limit: 2 sec / Memory Limit: 512 MB Score : 700 points Problem Statement

You are given N integers A 1 , A 2 , …, A N . Consider the sums of all non-empty subsequences of A . There are 2 N − 1 such sums, an odd number. Let the list of these sums in non-decreasing order be S 1 , S 2 , …, S 2 N − 1 . Find the median of this list, S 2 N − 1 . Constraints 1 ≤ N ≤ 2000 1 ≤ A i ≤ 2000 All input values are integers.

C-中位数/时间限制:2秒/内存限制:512 MB得分:700分问题陈述

给您N个整数A 1,A 2,…,A N。 考虑A的所有非空子序列的总和。 这样的和有2 N-1个,一个奇数。 令这些和以非降序排列的列表为S 1,S 2,…,S 2 N -1。 找到该列表的中位数S 2 N − 1。 约束1≤N≤2000 1≤A i≤2000所有输入值都是整数。

input:
3
1 2 1

output:
2

solution

/*
Atcoder,  agc020_c
题意:
+ 给出一个长为n的序列,考虑所有(2^n-1)个非空子序列的和
+ 令这些和升序排列,输出它们的中位数。
思路:
+ 将序列等价于集合S,子序列就是非空子集A。对于每个A,它的补集要么是空集,要么也在S中。
+ 将两个互补的子集看做一组,那么每组中至少一个子集>=sum/2,另一个子集<=sum/2。
+ 所以这个序列的中位数就是第一个大于等于sum/2的子集,所以可以用01背包来解。
+ 因为V=4e6,N=2e3,VN=8e9,会超时,因为是二进制所以可以用bitset优化。因为dp[i]=max(dp[i],dp[i-ai]),所以左移的操作就可以做到这个目的。
+ 比如101011,ai=3,101011<<3=101011000,101011000|101011=10111011就相当于用ai更新了一次。
*/
#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e6+10;
bitset<maxn>b;
int main(){
	int n;  cin>>n;
	int sum = 0;
	vector<int>a(n);
	for(int i = 0; i < n; i++){
		cin>>a[i];  sum += a[i];
	}
	
	b[0] = 1;
	for(int i = 0; i < n; i++)
		b |= (b<<a[i]);
	
	int ans = sum+1>>1;
	while(b[ans]==0)ans++;
	cout<<ans<<"\n";
	return 0;
}
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 技术工厂 设计师:CSDN官方博客 返回首页