Codeforces 722C(并查集 + 思维)

本文链接:http://www.cnblogs.com/Ash-ly/p/5932712.html

题目链接:http://codeforces.com/problemset/problem/722/C

思路:

　　题目给的操作数从第 1 个到第 N 个数是删除原数组中的一个数, 那么反过来从后往前就是增加原数组中的一个数, 每增加一个值, 那么就有四种情况: 第一种和前后都不连续, 即自成一个集合; 第二种:和前面的数连续, 即和前一个数在一个集合; 第三种和之后一个数连续, 即和之后的一个数在一个集合; 第四种既和前面一个数连续也和后面一个数连续,那么说明前后两个集合被这个数合并成一个集合. 然后合并的时候维护每个集合的元素和 sum, 利用 max 记录当前集合 sum 和新增集合的 sum 中的最大值.

代码：

 #include <bits/stdc++.h>

 typedef long long LL;
 const int MAXN = ;
 using namespace std;
 LL pre[MAXN + ], cnt[MAXN + ], a[MAXN + ], pos[MAXN + ], visit[MAXN + ], ans[MAXN + ];

 int Find(int x) { return x == pre[x] ? x : pre[x] = Find(pre[x]); }

 void mix(int x, int y) {
     int fx = Find(x), fy = Find(y);
     if(fx != fy) pre[fy] = fx, cnt[fx] += cnt[fy]; //合并两个集合的同时维护新集合的 sum
 }

 int main() {
     ios_base::sync_with_stdio(); cin.tie();
     int n; cin >> n;
     for(int i = ; i <= n; i++) {
         cin >> a[i];
         pre[i] = i, visit[i] = ;
     }
     for(int i = ; i < n; i++) cin >> pos[i];
     LL maxn = ;
     for(int i = n - ; i >= ; i--) {
         int tp = pos[i];
         cnt[tp] = a[tp], visit[tp] = , ans[i] = maxn;
         if(tp !=  && visit[tp - ]) mix(tp, tp - ); // 新增一个元素后形成的四种情况
         if(tp != n && visit[tp + ]) mix(tp, tp + );
         maxn = max(maxn, cnt[ Find(tp) ]);
     }
     for(int i = ; i < n; i++) cout << ans[i] << endl;
     return ;
 }