Jersey Politics POJ - 2454 (随机算法)

传送门

题意：给定一个长度为3*K的整数序列S，将其划分为等长（各为K）的三个子序列S1,S2,S3，要求其中至少两个子序列各自的加和大于500*K，求满足上述条件的一个划分（所有输入必有解）。

题解：首先采用贪心策略，对输入序列S进行降序排序，将排序后的前2*K个元素划分给S1和S2，不再考虑S3。

设S'=(S1,S2)，现在考虑采用随机化算法，第一种策略：循环地随机重排S'中的所有元素，进行条件检测，若满足则退出循环输出结果，否则继续循环。这种策略会造成大量浪费，因为会出现如下情况 ：随即重排一次后，S1和S2各自包含的元素没有发生变化，导致此次重排没有意义。此外对整个S'进行重排也很耗时。第二种策略：每次分别在S1和S2中各随机选取一个元素进行互换，进行条件检测。这种策略能保证绝大部分的重排是有效的(除非本次随机选取的两个元素恰好是上次随机选取的那两个元素)。

附上代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>

using namespace std;

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

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

int main()
{
    int k;
    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);
    int sum1=0,sum2=0;
    for(int i=0;i<2*k;i+=2){
        num1[i/2]=num[i];
        sum1+=num[i].nu;
    }
    for(int i=1;i<=2*k;i+=2){
        num2[i/2]=num[i];
        sum2+=num[i].nu;
    }
    while(sum1<=500*k||sum2<=500*k){
        int rd1=rand()%k;
        int rd2=rand()%k;
        sum1=sum1-num1[rd1].nu+num2[rd2].nu;
        sum2=sum2-num2[rd2].nu+num1[rd1].nu;
        swap(num1[rd1],num2[rd2]);
    }
    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;
}