SuperHyperMarket（优先队列+重载）

SuperHyperMarket（优先队列+重载）

具体见代码注释

 /* */
 #include <iostream>
 #include <cstring>
 #include <queue>
 #include <cstdio>
 #include <string>
 #include <cmath>
 #include <utility>
 #include <algorithm>
 using namespace std;
 typedef long long ll;
 const int maxn=1e5+;
 const int inf=0x3f3f3f3f;
 const double pi=acos(-1.0);

 int arr[maxn];
 struct node
 {
     int mem;//该队的人数
     int num;//该队的编号
     int b1,b2;//该队的最后两个人
     int pi;//下一个人计算的花费
     bool operator<(const node &a)const//重载，从小到达排，先出队小的
     {
         if( a.pi==pi ) return num>a.num;//编号小的
         return pi>a.pi;//花费小的
     }
 } p[maxn];
 priority_queue<node> q;
 int loc[maxn];//标记所要去的队

 int main()
 {
     int n,k;
     cin>>n>>k;
     for(int i=; i<=n; i++) scanf("%d",&arr[i]);
     for(int i=; i<=k; i++) //初始化
     {
         p[i].b1=p[i].b2=;//the last purchase people
         p[i].mem=p[i].pi=;
         p[i].num=i;//编号
         q.push(p[i]);
     }

     for(int i=; i<=n; i++)
     {
         node x=q.top();
         loc[i]=x.num;
         q.pop();
         x.mem++;
         x.b2=x.b1;
         x.b1=arr[i];
         if( x.mem== )
         {
             x.pi=*arr[i];//化除为乘防止精度缺失
         }
         else
         {
             x.pi=x.mem*(x.b1+x.b2);//化除为乘防止精度缺失，所以不除2
         }
         q.push(x);
     }
     for(int i=; i<=n; i++)
     {
         if(i==n) printf("%d\n",loc[i]);
         else printf("%d ",loc[i]);
     }
     return ;
 }