POJ2454 Jersey Politics

随机化算法。

亲身试验贪心会T。

所以排序，然后在前2*k个数中随机交换。

祝大家新年快乐。

#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int k,tmp,cnt1,cnt2;
struct node
{
    int id,s;
    bool operator<(const node &b)const{
        return s>b.s;
    }
}q[200];
int main()
{
    scanf("%d",&k);
    tmp=k*500;
    for(int i=1;i<=k*3;++i)
    {
        q[i].id=i;scanf("%d",&q[i].s);
    }
    int cnt=0;
    sort(q+1,q+1+3*k);
    while(1)
    {
        cnt1=cnt2=0;
        for(int i=1;i<=k;++i)
        {
            cnt1+=q[i].s;
        }
        for(int i=k+1;i<=k*2;++i)
        {
            cnt2+=q[i].s;
        }
        if(cnt1>tmp&&cnt2>tmp)
        {
            for(int i=1;i<=k*3;++i)
            cout<<q[i].id<<endl;
            return 0;
        }
        swap(q[rand()%k+1],q[rand()%k+k+1]);
        cnt++;
    }
    return 0;
}