第一道atcoder的题呀!pkuwc不怎么样的回来了ww,又要好好刷题啦,那么二月份就从这道题开始吧。
之前没怎么接触过bitset讲真qwq
题意:有n个数a1,a2,...,an,选任意个数可以得到2^n -1个和,求这些和从小到大排序后的中位数。
这题还可以(D题嘛我就又有点想不清楚了)。
首先有一点:答案一定是大于等于总和的二分之一,设总和为S,取了一些数总和为P,那么S-P一定也是存在的,若P<S/2,那么就存在一个大于S/2的数与之对应。
很容易想到一个dp,dp[i][j]表示取了前i个数,和为j是否可行。但这样时间不行,所以就要用到bitset了呀,bitset用来存每个和是否可能出现。不用担心重复出现两个一样的和,因为这样的话与它对应的和也有多个,所以对于每个x,要么取dp>>x,要么不取dp
综合一下就是dp |= (dp>>x)。这样时间复杂度是O(n^2*max(ai)/64)。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <bitset>
#define N 2020
using namespace std;
bitset<N*N> dp;
int n;
int main()
{
scanf("%d",&n);
int sum=0;
dp[0]=1;
for(int i=1;i<=n;i++){
int x;
scanf("%d",&x);
sum+=x;
dp |= (dp<<x);
}
int ans=(sum+1)/2;
while(!dp[ans]) ans++;
printf("%d\n",ans);
return 0;
}