Codeforces 458C Elections 线段树

使得0号的选票是最高的(不能有和0号相同)的最小花费

枚举0号的最终选票

那么已知0号最终选票,则有些人选票比0号大的,那些票都要买下来。

如果买完了还是达不到 最终选票,就从所有剩下的选票里找前k小的。

用线段树求前k小的数的和

#include
using namespace std;
typedef long long ll;
#define Max int(10000+10)
#define INF 0x3f3f3f3f
const int N=int (1e5+10);
int n,cost,cnt;
vector g[N],a[N];
struct node {
	int r,l,size,sum;
} tree[10004*4];

void build(int l,int r,int id) {
	tree[id].l=l;
	tree[id].r=r;
	tree[id].size=0;
	tree[id].sum=0;
	if(l==r)
		return ;
	int mid=(l+r)/2;
	build(l,mid,id*2);
	build(mid+1,r,id*2+1);
}

void update(int pos,int val ,int id) {
	if(tree[id].l==tree[id].r) {
		tree[id].size+=val;
		tree[id].sum+=pos*val;
		return ;
	}
	int mid=(tree[id].l+tree[id].r)/2;
	if(pos<=mid)
		update(pos,val,2*id);
	else
		update(pos,val,2*id+1);
	tree[id].size=tree[2*id].size+tree[2*id+1].size;
	tree[id].sum=tree[2*id].sum+tree[2*id+1].sum;

}

int query(int k,int id) {
	if(k<=0)
		return 0;
	if(tree[id].size<=k)
		return tree[id].sum;
	
	
	if(tree[id].l==tree[id].r)
		return min(k,tree[id].size)*tree[id].l;
	int mid=(tree[id].l+tree[id].r)/2;
	if(tree[2*id].size<=k)
		return tree[2*id].sum+query(k-tree[2*id].size,2*id+1);
	else
		return query(k,2*id);
}

int work(int h) {
	for(int i=0; i=0; i--) {
			int ans=work(i);
			if(ans==-1)
				break;
			sum=min(sum,ans);
		}

		printf("%d\n",sum);
	}

	return 0;
}
/*
5
1 2
1 2
1 2
2 1
0 0

3
*/

 

你可能感兴趣的:(ACM,线段树)