随机化算法+贪心!
将3*k排序后分成3分,将第二第三份的和分别加起来,让和与500*k比较,都大于则输出,否则,随机生成2个数,在第二第三份中交换!
代码如下:
#include<iostream> #include<stdio.h> #include<cmath> #include<algorithm> using namespace std; struct an { int w,lab; }p[200]; bool cmp(const an &a,const an &b) { return a.w<b.w; } int main() { int i,j,n,sa,sb,ans,a,b; cin>>n; for(i=0;i<3*n;i++){ cin>>p[i].w; p[i].lab=i+1; } sort(p,p+3*n,cmp); sa=0;sb=0; for(i=n;i<n+n;i++) sa+=p[i].w; for(i=n+n;i<3*n;i++) sb+=p[i].w; ans=n*500; bool flag=0; if(sa>ans&&sb>ans) flag=1; while(!flag){ a=rand()%n+n; b=rand()%n+2*n; sa=sa-p[a].w+p[b].w; sb=sb-p[b].w+p[a].w; swap(p[a],p[b]); if(sa>ans&&sb>ans){ flag=1; break; } } for(i=0;i<3*n;i++) cout<<p[i].lab<<endl; return 0; }