ZOJ-3715-Kindergarten Election

在学期开始的幼儿园，n个小孩（1到n）在课堂上需要选择他们的新领导。
第一个孩子会投票给他最好的朋友fi（其中1≤fi≤n，这太可惜了，自己投票，所以fi≠i）。 

获得最多票数的孩子将是领袖。 如果有一个以上的孩子获得最多的票数，在新学期将有多个领导人。

小谢尔顿（下标为1）想成为唯一的领导者。 （这意味着他得到的票数应该严格大于任何其他。）

如果他给ci糖果给第i个孩子，第i个孩子会认为谢尔顿作为新的最好的朋友，当然投票谢尔顿。

请帮助谢尔顿成为唯一的领导者，最低的糖果成本。 顺便说一下，谢尔顿应该投票支持自己想要的任何一个人。

枚举获胜所需要的票数，从其他超过该票数的中买票（当然从便宜的开始买），如果买完后还不够枚举的票数就买其他没有买过的票.

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <queue>
#include <vector>
#include <math.h>
#include <stack>
#include <map>
#define rtl rt<<1
#define rtr rt<<1|1
typedef long long LL;
using namespace std;
const int MAX = 15000+10;
const double eps = 1e-10;
const double PI = acos(-1.0);
int t, n, f[MAX], c[MAX], ans, ant, num, num0;
bool vis[MAX];
struct node
{
int c, id;
};
bool operator < (node a, node b)
{
if(a.c!=b.c)return a.c>b.c;
return a.id<b.id;
}
priority_queue<node>que[MAX], que2[MAX], qq, qqq;
int main()
{
scanf("%d", &t);
while(t--)
{
scanf("%d", &n);
for(int i = 1; i<=n; ++i)
while(!que[i].empty())que[i].pop();
while(!qq.empty())qq.pop();
for(int i = 2; i<=n; ++i)
cin>>f[i];
num0 = 0;
for(int i = 2; i<=n; ++i)
{
node ver;
scanf("%d", &ver.c);
ver.id = i;
if(f[i]!=1)
{
que[f[i]].push(ver);
qq.push(ver);
}
else num0++;
}
ans = 0x7fffffff;
for(int i = max(2, (int)que[1].size()); i<=n; ++i)
{
num = num0;
ant = 0;
memset(vis, 0, sizeof(vis));
for(int j = 2; j<=n; ++j)
{
que2[j] = que[j];
while((int)que2[j].size()>i-1)
{
node ver = que2[j].top();
que2[j].pop();
ant+=ver.c;
vis[ver.id] = 1;
num++;
}
if(num>i)break;
}
if(num<i)
{
qqq = qq;
while(num<i)
{
node ver = qqq.top();
qqq.pop();
if(!vis[ver.id])
{
ant+=ver.c;
num++;
}
}
}
if(num==i)ans = min(ans, ant);
}
if(ans==0x7fffffff)ans = 0;
cout<<ans<<endl;
}
return 0;
}