Codeforces 798D

  这两天后缀数组整多了整点有意思的，随机算法。

题意给你两个数组，让你求一个排列使得这个排列对应的两个数组前n/2+1个数之和的二倍大于每个序列总和。

下面先贴下这题正解

二维贪心，按a从大到小排，把第一个拿着，以后每两个里面选一个a的性质可以满足。

b的话就在两个里面选一个b较大的，这样可以保证有一半的b都大于另一半的b。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long ll;
int n,ans[N],cnt;
struct node
{
    int a,b,id;
    bool operator <(const node &b)const{
        return a>b.a;
    }
}q[N];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)scanf("%d",&q[i].a),q[i].id=i;
    for(int i=1;i<=n;++i)scanf("%d",&q[i].b);
    sort(q+1,q+1+n);
    ans[++cnt]=q[1].id;
    for(int i=2;i<=n;i+=2)
    {
        if(q[i].b>q[i+1].b)
        ans[++cnt]=q[i].id;
        else ans[++cnt]=q[i+1].id;
    }
    cout<<cnt<<endl;
    for(int i=1;i<=cnt;++i)
    cout<<ans[i]<<" ";
    return 0;
}

下面就是随机化了。。。祝大家新年快乐。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long ll;
int n,ans[N],cnt;
ll sa,sb,ssa,ssb;
struct node
{
    int a,b,id;
}q[N];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)scanf("%d",&q[i].a),q[i].id=i,sa+=q[i].a;
    for(int i=1;i<=n;++i)scanf("%d",&q[i].b),sb+=q[i].b;
    while(1)
    {
        random_shuffle(q+1,q+1+n);
        ssa=ssb=0;
        for(int i=1;i<=n/2+1;++i){
            ssa+=q[i].a;ssb+=q[i].b;
        }
        if(ssa*2>sa&&ssb*2>sb){
            cout<<n/2+1<<endl;
            for(int i=1;i<=n/2+1;++i)cout<<q[i].id<<" ";
            return 0;
        }
    }
    return 0;
}