POJ 2454 Jersey Politics 分组问题 随机化算法

题意：给出3*k个数，将它们分成三分，每份k个数，要求至少存在有两份，使得每一份的k个数的和大于500*k。

思路：

先从大到小的进行排序，然后只需要将前2*n大的数进行随机的分配成两组就行了。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <map>
#include <queue>
#define LL long long
#define INF 0x3f3f3f3f
#define bug puts("***************")
using namespace std;
const int N =1100;
struct node{
    int id,val;
    friend bool operator <(node p1,node p2){
        return p1.val>p2.val;
    }
}a[N*3];;
int main(){
    int n;
    srand((unsigned)time(NULL));
    scanf("%d",&n);
    for(int i=0;i<3*n;i++){
        scanf("%d",&a[i].val);
        a[i].id=i;
    }
    sort(a,a+n*3);
    int s1=0,s2=0;
    for(int i=0;i<n;i++){
        s1+=a[i].val;
    }
    for(int i=n;i<2*n;i++){
        s2+=a[i].val;
    }
    while(s1<=500*n||s2<=500*n){
        int k1=rand()%n;
        int k2=rand()%n+n;
        s1=s1-a[k1].val+a[k2].val;
        s2=s2-a[k2].val+a[k1].val;
        swap(a[k1],a[k2]);
    }
    for(int i=0;i<3*n;i++){
        printf("%d\n",a[i].id+1);
    }
    return 0;
}