poj2454--Jersey Politics（随机化算法）

题目链接：点击打开链接

题目大意：给出3*n个地方，每个地方有1000票，现在有每个地方的得票数，问如何分成3个区域（每个区域n个城市），使得每个区域的总票数过半数。

首先对于所有地方由大到小排序，选择前2*n个，划分到两个区域，这样是能得到和最大的方法，然后随机交换两个不同区域的城市，一直到这两个区域的和都大于3*500

#include <cstdio>
#include <cstring>
#include <time.h>
#include <algorithm>
using namespace std ;
struct node{
    int c , id ;
}p[200] ;
int cmp(node a,node b) {
    return a.c > b.c ;
}
int a[70] , b[70] , c[70] , sum1 , sum2 ;
int main() {
    //srand( (unsigned)time( NULL ) ) ;
    int n , m , i , j ;
    scanf("%d", &n) ;
    for(i = 0 ; i < 3*n ; i++) {
        scanf("%d", &p[i].c) ;
        p[i].id = i ;
    }
    sort(p,p+3*n,cmp) ;
    sum1 = sum2 = 0 ;
    for(i = 0 ; i < n ; i++) {
        a[i] = i ;
        sum1 += p[i].c ;
        b[i] = i+n ;
        sum2 += p[i+n].c ;
        c[i] = i+2*n ;
    }
    m = 500*n ;
    while( sum1 <= m || sum2 <= m ) {
        i = rand()%n ;
        j = rand()%n ;
        sum1 -= p[ a[i] ].c ;
        sum1 += p[ b[j] ].c ;
        sum2 += p[ a[i] ].c ;
        sum2 -= p[ b[j] ].c ;
        swap(a[i],b[j]) ;
    }
    for(i = 0 ; i < n ; i++) printf("%d\n", p[a[i]].id+1) ;
    for(i = 0 ; i < n ; i++) printf("%d\n", p[b[i]].id+1) ;
    for(i = 0 ; i < n ; i++) printf("%d\n", p[c[i]].id+1) ;
    return 0 ;
}