ural 1144. The Emperor's Riddle

来源：互联网发布：淘宝权女友辣椒和静雯编辑：程序博客网时间：2024/04/27 23:02

06年的论文题

玄学！！！！

重要的事情重复O(玄学)遍

写完这题我整个人都不好了

一开始由于很无脑地用vector，结果死活卡在第7个点，无限TLE。

没办法看了下别人的，卧槽贪心+玄学？果然纯玄学过不去啊

于是改了一下，顺便优化了下常数就过了。

再也不写玄学题了QAQ

#include<iostream>#include<cstdio>#include<cstring>#include<vector>#include<algorithm>#include<queue>#include<ctime>#include<cstdlib>#define mk(i,j) make_pair(i,j)using namespace std;const int N=10000+5;const int M=1000+5;int v[N],rk[M],to[M],tmp[N],s[2][M],sum[M],belong[N],tag[N];priority_queue<pair<int,int> >gold,gen;int n,m;int check(){for(int i=1;i<=m;i++)sum[i]=0;for(int i=1;i<=n;i++)sum[belong[i]]+=v[i];int mx=sum[1],mi=sum[1];for(int i=2;i<=m;i++)mx=max(mx,sum[i]),mi=min(mi,sum[i]);return mx-mi;}bool cmp1(int a,int b){return s[0][a]<s[0][b];}bool cmp2(int a,int b){return s[1][a]>s[1][b];}vector<int>mag[M];int main(){//freopen("a.in","r",stdin);srand(time(0));int k;scanf("%d%d%d",&n,&m,&k);for(int i=1;i<=n;i++)scanf("%d",&v[i]);for(int i=1;i<=m;i++)gen.push(mk(-sum[i],i));for(int i=1;i<=n;i++)gold.push(mk(v[i],i));for(int i=1;i<=n;i++){int u=gen.top().second,t=gold.top().second;gen.pop();gold.pop();sum[u]+=v[t];belong[t]=u;gen.push(mk(-sum[u],u));}int ans;while((ans=check())>k){for(int i=1;i<=n;i++)tag[i]=rand()&1;for(int t=0;t<2;t++)for(int i=1;i<=m;i++)s[t][i]=0;for(int i=1;i<=n;i++)s[tag[i]][belong[i]]+=v[i];for(int t=0;t<2;t++){for(int i=1;i<=m;i++)rk[i]=i;if(t==0)sort(rk+1,rk+1+m,cmp1);else sort(rk+1,rk+1+m,cmp2);for(int i=1;i<=m;i++)to[rk[i]]=i;for(int i=1;i<=n;i++)if(tag[i]==t)tmp[i]=to[belong[i]];}for(int i=1;i<=n;i++)belong[i]=tmp[i];}printf("%d\n",ans);for(int i=1;i<=n;i++)mag[belong[i]].push_back(i);for(int i=1;i<=m;i++){for(int j=0;j<mag[i].size();j++)printf("%d ",mag[i][j]);putchar('\n');}return 0;}

0 0