思路: 枚举当选时候的票数x。 则如果其他人所得的票数大于x则需要把他减少到x-1,减少的时候注意要选择需要糖果最少的。如果每人减少下来的加到1号上所得的票数已经超过x则该方案不可行。如果小于x则从全体中选出最小的 填满x。 要注意1号要把手中的票投给任何一个,所以的保证其他人中至少有一个得票数小于x-1.
代码:
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include <iostream>
using namespace std;
struct g
{
int id,c;
}s[105],ss[105];
int a[105][105];
int f[105],fa[105],h[105];
bool cmp(int x,int y)
{
return s[x].c>s[y].c;
}
bool cmp1(g x, g y)
{
return x.c<y.c;
}
int main()
{
int i,tt,cnt,sum,ans,count,j,k,n,m;
scanf("%d",&tt);
while(tt--)
{
cnt=count=0;
memset(h,0,sizeof(h));
scanf("%d",&n);
for(i=2;i<=n;i++)
{
scanf("%d",&f[i]);
a[f[i]][h[f[i]]++]=i;
if(f[i]==1)
count++;
}
for(i=2;i<=n;i++)
{
s[i].id=i;
scanf("%d",&s[i].c);
ss[i]=s[i];
}
for(i=1;i<=n;i++)
sort(a[i],a[i]+h[i],cmp);
sort(ss+2,ss+n+1,cmp1);
ans=1000000000;
for(i=count;i<n;i++)
{
for(j=1;j<=n;j++)
fa[j]=f[j];
cnt=i-count;
sum=0;
for(j=2;j<=n;j++)
{
int tmp=h[j];
while(tmp!=0&&tmp>=i)
{
sum+=s[a[j][tmp-1]].c;
fa[a[j][tmp-1]]=1;
tmp--;
cnt--;
}
}
if(cnt<0)continue;
if(i+(n-1)*(i-1)==n)//其他人至少需要一个小于i-1票。
cnt=1;
int ri=2;
while(cnt)
{
while(fa[ss[ri].id]==1)
ri++;
sum+=s[ss[ri].id].c;
cnt--;
ri++;
}
if(sum<ans)
ans=sum;
}
printf("%d\n",ans);
}
}