题意:1号小朋友想当唯一的班长,求他贿赂其他小朋友所需的最少糖果数。n个小朋友,接下来是(n-1)个小朋友想投票的对象,接下来是收买(n-1)个小朋友分别所需花费。
思路:先统计每个小朋友的得票数,再枚举1号小朋友可能的得票数,更新出他当上班长所需的最少花费,直到枚举到最优解。
#include<stdio.h>
#include<string.h>
#include<cmath>
#include<algorithm>
#define eps 1e-5
#define MM 0x3f3f3f3f
using namespace std;
int vis[505],f[505],c[505],sum[505],id[505];
int cmp(int a,int b)
{
return c[a]<c[b];
}
int main()
{
int T,n,i,j,k,cnt;
scanf("%d",&T);
while (T--)
{
scanf("%d",&n);
memset(sum,0,sizeof(sum));
for (i=2;i<=n;i++) scanf("%d",&f[i]),sum[f[i]]++;//统计各小朋友的得票数
for (i=2;i<=n;i++) scanf("%d",&c[i]);
cnt=MM;
for (k=sum[1];k<n;k++)//枚举1号小朋友的得票数
{
int shu=0,num,zhan=sum[1];//shu为花费数,zhan为当前1号小朋友的票数
memset(vis,0,sizeof(vis));
for (i=2;i<=n;i++)//将1号与其他小朋友得票数比较
if (sum[i]>=k)//若比1号票多,则将一部分转给1号
{
num=0;
for (j=2;j<=n;j++)
if (f[j]==i) id[++num]=j;
sort(id+1,id+1+num,cmp);
for (j=1;j<=sum[i]-(k-1);j++) shu+=c[id[j]],vis[id[j]]=1;
zhan+=sum[i]-(k-1);
}
//枚举结果非最优解则继续枚举,k增加了即为非最优解
num=0;
for (j=2;j<=n;j++)//若1号票数未达到枚举值,补满
if (!vis[j]) id[++num]=j;
sort(id+1,id+1+num,cmp);
for (i=1;i<=k-zhan;i++) shu+=c[id[i]];//达到当前枚举值所需花费
cnt=min(cnt,shu);
}
printf("%d\n",cnt);
}
return 0;
}