[bzoj2288][POJ Challenge]生日礼物

Posted 2020-06-13

　　用堆维护双向链表来贪心。。。

　　数据范围显然不容许O(nm)的傻逼dp>_<。。而且dp光是状态就n*m个了。。显然没法优化

　　大概就会想到贪心乱搞了吧。。。一开始想贪心地通过几段小的负数把正数连接成一段，但到底是要连接在一起还是直接扔掉不好判断

　　然后就跑去翻题解了。。。题解讲的挺好的，连我都看懂了>_<。。题解网址：http://www.cnblogs.com/tuigou/p/4868127.html

　　虽然选正数和负数的意义不同，但实际的操作都是把两边的数合并起来。还有就是，对于在左端或右端的负数，把它删去后并不会减少当前选取的段数。

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<queue>
 6 #include<cstdlib>
 7 #include<cmath>
 8 #define ll long long
 9 using namespace std;
10 const int maxn=100233;
11 struct zs{
12     int id;
13 };
14 priority_queue <zs>q;
15 int a[maxn],cnt,pre[maxn<<1],next[maxn<<1],v[maxn<<1];
16 int i,j,n,m,zsnum,ans;
17 bool del[maxn<<1];
18 
19 int ra,fh;char rx;
20 inline int read(){
21     rx=getchar(),ra=0,fh=1;
22     while((rx<‘0‘||rx>‘9‘)&&rx!=‘-‘)rx=getchar();
23     if(rx==‘-‘)fh=-1,rx=getchar();
24     while(rx>=‘0‘&&rx<=‘9‘)ra*=10,ra+=rx-48,rx=getchar();return ra*fh;
25 }
26 bool operator <(zs a,zs b){return abs(v[a.id])>abs(v[b.id]);}
27 int main(){
28     n=read(),m=read();if(!m){puts("0");return 0;}
29     for(i=1;i<=n;i++)a[i]=read();cnt=0;
30     for(i=1;i<=n;v[cnt]+=a[i++])
31         if((ll)a[i]*(ll)v[cnt]<0||!cnt)cnt++;
32     for(i=1;i<=cnt;i++)if(v[i]>0)zsnum++,ans+=v[i];
33     if(zsnum>m){
34         for(i=1;i<=cnt;i++)q.push((zs){i}),pre[i]=i-1,next[i]=i+1;//,printf("  %d",v[i]);puts("");
35         pre[1]=next[cnt]=0;
36         for(i=zsnum-m;i;i--){
37             while(!q.empty()&&del[q.top().id])q.pop();if(q.empty())break;
38             int x=q.top().id,pr=pre[x],nex=next[x];
39             
40             q.pop(),ans-=abs(v[x]),del[x]=1;
41             if(!(pr&&nex)){
42                 if(v[x]<0)i++,ans+=abs(v[x]);
43                 if(pr)next[pr]=0;if(nex)pre[nex]=0;
44             }
45             else{
46                 del[pr]=del[nex]=1;
47                 v[++cnt]=v[pr]+v[x]+v[nex];
48                 q.push((zs){cnt});
49                 if(pre[pr])pre[cnt]=pre[pr],next[pre[cnt]]=cnt;
50                 if(next[nex])next[cnt]=next[nex],pre[next[cnt]]=cnt;
51             }
52         }
53     }
54     printf("%d\\n",ans);
55     return 0;
56 }

View Code