标签：ch P1484 read ll pq maxn 种树 define

 1 #include <bits/stdc++.h>
 2 #define _for(i,a,b) for(int i = (a);i < b;i ++)
 3 #define _rep(i,a,b) for(int i = (a);i > b;i --)
 4 #define INF 0x3f3f3f3f
 5 #define MOD 1000000007
 6 #define pb push_back
 7 #define maxn 500003 
 8 typedef long long ll;
 9 using namespace std;
10 inline ll read()
11 {
12     ll ans = 0;
13     char ch = getchar(), last = ' ';
14     while(!isdigit(ch)) last = ch, ch = getchar();
15     while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar();
16     if(last == '-') ans = -ans;
17     return ans;
18 }
19 inline void write(ll x)
20 {
21     if(x < 0) x = -x, putchar('-');
22     if(x >= 10) write(x / 10);
23     putchar(x % 10 + '0');
24 }
25 ll N,K;
26 struct Tree
27 {
28     ll id;
29     ll val;
30 };
31 struct cmp
32 {
33     bool operator() (const Tree a,const Tree b) const
34     {
35         return a.val < b.val;
36     }
37 };
38 priority_queue<Tree,vector<Tree>,cmp> pq;
39 ll a[maxn];
40 ll l[maxn],r[maxn];
41 ll vis[maxn];
42 ll ans = 0;
43 int main()
44 {
45     N = read();
46     K = read();
47     _for(i,1,N+1)
48     {
49         a[i] = read();
50         l[i] = i-1;r[i] = i+1;
51         pq.push({i,a[i]});
52     }
53     
54     _for(i,0,K)
55     {
56         Tree tmp = pq.top();
57         while(vis[tmp.id])
58         {
59             pq.pop();
60             tmp = pq.top();
61         }
62         if(tmp.val<=0)
63             break;
64         pq.pop();
65         ans += tmp.val;
66         pq.push({tmp.id,a[l[tmp.id]]+a[r[tmp.id]]-tmp.val});
67         a[tmp.id] = a[l[tmp.id]]+a[r[tmp.id]]-tmp.val;
68         vis[l[tmp.id]] = vis[r[tmp.id]] = 1;
69         l[tmp.id]=l[l[tmp.id]];r[l[tmp.id]]=tmp.id;
70         r[tmp.id]=r[r[tmp.id]];l[r[tmp.id]]=tmp.id;
71     }
72     write(ans);
73     return 0;
74 }

标签：ch,P1484,read,ll,pq,maxn,种树,define
来源： https://www.cnblogs.com/Asurudo/p/11564116.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。