zoj 3715 Kindergarten Election

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5018

题意:

有n个学生要选出班长编号从1到n,1号人比较虚荣，想自己当班长，每个人都可以投一票给自己心目中的人但不能投给自己，只要谁的票数最高谁就可以当班长。给出每个人心目中的投票人，以及贿赂每个人所需要的糖果，（只要你给了那个人一定数目的糖果他就会支持你） 。问1号如果当班长的话，最少需要的糖果数。

思路：

由于这里的n比较小，我们只要枚举1当班长时的得票数x，然后再将其他人的得票数大于x的变为x-1  (减少的给1并且减少的肯定是所需糖果树最少的)，然后检查最后1的得票数，如果大于x那么肯定无解，如果等于x，只要保证2到n中有得票数<= x - 2的即可。如果小于x,那么从剩下没有支持1的中，找出所需糖果树最少的来贿赂得票知道等于x为止。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>
#include <algorithm>
#include <string>
#include <set>
#include <functional>
#include <numeric>
#include <sstream>
#include <stack>
#include <map>
#include <queue>

#define CL(arr, val) memset(arr, val, sizeof(arr))

#define lc l,m,rt<<1
#define rc m + 1,r,rt<<1|1
#define pi acos(-1.0)
#define L(x)    (x) << 1
#define R(x)    (x) << 1 | 1
#define MID(l, r)   (l + r) >> 1
#define Min(x, y)   (x) < (y) ? (x) : (y)
#define Max(x, y)   (x) < (y) ? (y) : (x)
#define E(x)        (1 << (x))
#define iabs(x)     (x) < 0 ? -(x) : (x)
#define OUT(x)  printf("%I64d\n", x)
#define lowbit(x)   (x)&(-x)
#define Read()  freopen("din.txt", "r", stdin)
#define Write() freopen("dout.txt", "w", stdout)


#define M 137
#define N 115

using namespace std;


const int inf = 0x7f7f7f7f;
const int mod = 1000000007;

int ans,res;
int vote[N],c[N],b[N];
map<int,int>mp;
bool vt[N],vt2[N];
int n;

int getR(int k)
{
    for (int i = 2; i <= n; ++i)
    {
        if (b[i] >= k)
        {
           while (b[i] != k - 1)
           {
               int t1 = 0, t2 = inf;
               for (int j = 2; j <= n; ++j)
               {
                   if (!vt2[j] && mp[j] == i && c[j] < t2)
                   {
                       t1 = j;
                       t2 = c[j];
                   }
               }
               ans += t2; vt2[t1] = true;
               b[i]--; b[1]++;
           }
        }
    }
    return b[1];
}
int main()
{
//    Read();
    int T,i,j;
    int x;
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        CL(vote,0); mp.clear();
        CL(vt,false);

        for (i = 2; i <= n; ++i)
        {
            scanf("%d",&x);
            mp[i] = x;  vote[x]++;//x的得票数
            if (x == 1) vt[i] = true;//i是否支持1
        }
        for (i = 2; i <= n; ++i) scanf("%d",&c[i]);

        res = inf;
        for (int k = vote[1]; k <= n; ++k)
        {
            if (k <= 1) continue;

            for (i = 1; i <= n; ++i) b[i] = vote[i];
            for (i = 1; i <= n; ++i) vt2[i] = vt[i];
            ans = 0;

            int no = getR(k);

            if (no == k)
            {
                for (j = 2; j <= n; ++j)//保证存在一个得票数小于等k - 2的来结束1的票
                {
                    if (b[j] <= k - 2) break;
                }
                if (j <= n)
                {
                    res = min(res,ans);
                }
            }
            else if (no < k)
            {
                while (no < k)//从没有支持1的人中选出所需糖果数最少的人贿赂得票
                {
                    int t1 = 0,t2 = inf;
                    for (i = 2; i <= n; ++i)
                    {
                        if (!vt2[i] && c[i] < t2)
                        {
                            t1 = i; t2 = c[i];
                        }
                    }
                    vt2[t1] = true;
                    ans += t2;
                    no++;
                }
                res = min(res,ans);
            }
        }
        printf("%d\n",res);
    }
    return 0;
}