2288: 【POJ Challenge】生日礼物

2288: 【POJ Challenge】生日礼物

https://lydsy.com/JudgeOnline/problem.php?id=2288

分析：

　　贪心+堆+链表。

　　首先把序列变一下，把相邻的同符号的合并起来，让序列的第一个是整数，最后一个也是整数。

　　如果直接算最大的选的不好算，那么考虑算最小的不选的，正难则反。

　　然后把所有的整数都加起来，这就是最大的共和。如果此时的段数<=m，那么直接输出就好了。否则，需要选几个数字，来合并它左右的段，以此是段的总数-1。

　　选的是一个负数，表示，左边的段后右边的段经过这个负数合起来了。选的是一个正数，那么表示这个正数不要了。由于减去的都是绝对值的大小，那么可以把序列变成正的。此时问题转化为，选k个数，使得它们的和最小，并且不能相邻。

　　然后用堆+链表维护。类似上一道题。

代码：

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<iostream>
 #include<cmath>
 #include<cctype>
 #include<set>
 #include<queue>
 #include<vector>
 #include<map>
 using namespace std;
 typedef long long LL;

 inline int read() {
     int x=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
     for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
 }

 const int N = ;
 const int INF = 1e9;

 struct Node{
     int x, id;
     Node() {}
     Node(int a,int b) { x = a, id = b; }
     bool operator < (const Node &A) const {
         return x > A.x;
     }
 };
 priority_queue< Node > q;
 int pre[N], nxt[N], vis[N], a[N], b[N];
 int cnt;

 void del(int x) {
     if (!x || x == cnt + ) return ;
     vis[x] = ;
     b[x] = INF;
     nxt[pre[x]] = nxt[x];
     pre[nxt[x]] = pre[x];
 }

 int main() {
     int n = read(), m = read(); cnt = ;

     b[] = a[] = read();
     int last = a[];
     for (int i = ; i <= n; ++i) {
         a[i] = read();
         if (a[i] == ) continue;
         if (a[i] * last < ) b[++cnt] = a[i];
         else b[cnt] += a[i];
         last = a[i];
     }
     if (b[cnt] < ) cnt --;
     if (b[] < ) { cnt --; for (int i = ; i <= cnt; ++i) b[i] = b[i + ]; }
     int ans = , d = ;
     for (int i = ; i <= cnt; ++i) {
         if (b[i] > ) ans += b[i], d ++;
         else b[i] = -b[i];
         q.push(Node(b[i], i));
     }
     if (d <= m) { cout << ans << " "; return ; }
     else d = d - m;

     b[] = INF, b[cnt + ] = INF;
     for (int i = ; i <= cnt + ; ++i) nxt[i] = i + , pre[i] = i - ;
     pre[] = ; nxt[cnt + ] = cnt + ;

     while (d --) {
         while (vis[q.top().id]) q.pop();
         Node now = q.top(); q.pop();
         int i = now.id; ans -= now.x;
         int t = min(INF, b[pre[i]] + b[nxt[i]] - now.x);
         del(pre[i]), del(nxt[i]);
         b[i] = t; q.push(Node(b[i], i));
     }
     cout << ans;
     return ;
 }