SuperHyperMarket（优先队列+重载）

Posted 2020-11-15 wsy107316

SuperHyperMarket（优先队列+重载）

具体见代码注释

 1 /* */
 2 #include <iostream>
 3 #include <cstring>
 4 #include <queue>
 5 #include <cstdio>
 6 #include <string>
 7 #include <cmath>
 8 #include <utility>
 9 #include <algorithm>
10 using namespace std;
11 typedef long long ll;
12 const int maxn=1e5+10;
13 const int inf=0x3f3f3f3f;
14 const double pi=acos(-1.0);
15 
16 int arr[maxn];
17 struct node
18 {
19     int mem;//该队的人数
20     int num;//该队的编号
21     int b1,b2;//该队的最后两个人
22     int pi;//下一个人计算的花费
23     bool operator<(const node &a)const//重载，从小到达排，先出队小的
24     {
25         if( a.pi==pi ) return num>a.num;//编号小的
26         return pi>a.pi;//花费小的
27     }
28 } p[maxn];
29 priority_queue<node> q;
30 int loc[maxn];//标记所要去的队
31 
32 int main()
33 {
34     int n,k;
35     cin>>n>>k;
36     for(int i=1; i<=n; i++) scanf("%d",&arr[i]);
37     for(int i=1; i<=k; i++) //初始化
38     {
39         p[i].b1=p[i].b2=0;//the last purchase people
40         p[i].mem=p[i].pi=0;
41         p[i].num=i;//编号
42         q.push(p[i]);
43     }
44 
45     for(int i=1; i<=n; i++)
46     {
47         node x=q.top();
48         loc[i]=x.num;
49         q.pop();
50         x.mem++;
51         x.b2=x.b1;
52         x.b1=arr[i];
53         if( x.mem==1 )
54         {
55             x.pi=2*arr[i];//化除为乘防止精度缺失
56         }
57         else
58         {
59             x.pi=x.mem*(x.b1+x.b2);//化除为乘防止精度缺失，所以不除2
60         }
61         q.push(x);
62     }
63     for(int i=1; i<=n; i++)
64     {
65         if(i==n) printf("%d
",loc[i]);
66         else printf("%d ",loc[i]);
67     }
68     return 0;
69 }