bzoj1732[usaco2005 feb]Jersey Politics 牛的政治 随机化

#include <set>
#include <map>
#include <queue>
#include <math.h>
#include <vector>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>

#define eps 1e-8
#define pi acos(-1.0)
#define inf 107374182
#define inf64 1152921504606846976
#define lc l,m,tr<<1
#define rc m + 1,r,tr<<1|1
#define iabs(x)  ((x) > 0 ? (x) : -(x))
#define clear1(A, X, SIZE) memset(A, X, sizeof(A[0]) * (SIZE))
#define clearall(A, X) memset(A, X, sizeof(A))
#define memcopy1(A , X, SIZE) memcpy(A , X ,sizeof(X[0])*(SIZE))
#define memcopyall(A, X) memcpy(A , X ,sizeof(X))
#define max( x, y )  ( ((x) > (y)) ? (x) : (y) )
#define min( x, y )  ( ((x) < (y)) ? (x) : (y) )

using namespace std;

struct node
{
int nu,t;
}num[200],num1[100],num2[100],temp;

bool cmp(node a,node b)
{
return a.nu>b.nu;
}

int main()
{
int k,sum1=0,sum2=0,rd1,rd2;
scanf("%d",&k);
for(int i=0;i<3*k;i++)
{
scanf("%d",&num[i].nu);
num[i].t=i;
}
sort(num,num+(3*k),cmp);
for(int i=0;i<2*k;i+=2)
{
//printf("%d\n",num[i].t+1);
num1[i/2]=num[i];
sum1+=num[i].nu;
}
for(int i=1;i<2*k;i+=2)
{
//printf("%d\n",num[i].t+1);
num2[i/2]=num[i];
sum2+=num[i].nu;
}
while(sum1<=500*k||sum2<=500*k)
{
rd1=rand()%k;
rd2=rand()%k;
sum1=sum1-num1[rd1].nu+num2[rd2].nu;
sum2=sum2-num2[rd2].nu+num1[rd1].nu;
temp=num1[rd1];
num1[rd1]=num2[rd2];
num2[rd2]=temp;
}
for(int i=0;i<k;i++)
{
printf("%d\n",num1[i].t+1);
}
for(int i=0;i<k;i++)
{
printf("%d\n",num2[i].t+1);
}
for(int i=2*k;i<3*k;i++)
{
printf("%d\n",num[i].t+1);
}
return 0;
}