Codeforces 798D

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

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

下面先贴下这题正解

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

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

 #include<bits/stdc++.h>
 using namespace std;
 const int N=1e5+;
 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=;i<=n;++i)scanf("%d",&q[i].a),q[i].id=i;
     for(int i=;i<=n;++i)scanf("%d",&q[i].b);
     sort(q+,q++n);
     ans[++cnt]=q[].id;
     for(int i=;i<=n;i+=)
     {
         if(q[i].b>q[i+].b)
         ans[++cnt]=q[i].id;
         else ans[++cnt]=q[i+].id;
     }
     cout<<cnt<<endl;
     for(int i=;i<=cnt;++i)
     cout<<ans[i]<<" ";
     return ;
 }

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

 #include<bits/stdc++.h>
 using namespace std;
 const int N=1e5+;
 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=;i<=n;++i)scanf("%d",&q[i].a),q[i].id=i,sa+=q[i].a;
     for(int i=;i<=n;++i)scanf("%d",&q[i].b),sb+=q[i].b;
     while()
     {
         random_shuffle(q+,q++n);
         ssa=ssb=;
         for(int i=;i<=n/+;++i){
             ssa+=q[i].a;ssb+=q[i].b;
         }
         if(ssa*>sa&&ssb*>sb){
             cout<<n/+<<endl;
             for(int i=;i<=n/+;++i)cout<<q[i].id<<" ";
             return ;
         }
     }
     return ;
 }