Codeforces #722C: Destroying Array 题解

这题我看到以后想到的最显然的做法是set

对数组做一个前缀和

先开一个set来存各个线段的开始和结束，再开一个multiset(一定要是multiset因为和可能会有重复)存所有线段的和

当删除一个点x时，先在set1中找到这个点所在的线段的左右端点，求出这个线段的原本和，在set2中删除，然后把l~x, x~r压入set1,set2

记得开long long

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <utility>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <queue>
#include <deque>
#define x first
#define y second
#define mp make_pair
#define pb push_back
#define LL long long
#define Pair pair<int,int>
#define LOWBIT(x) x & (-x)
using namespace std;

const int MOD=1e9+7;
const int INF=0x7ffffff;
const int magic=348;

int n;
set<int> s;
multiset<LL> ss;
int a[100048],seq[100048];
LL sum[100048];

int main ()
{
	int i;
	scanf("%d",&n);
	for (i=1;i<=n;i++) scanf("%d",&a[i]);
	sum[0]=0;
	for (i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
	for (i=1;i<=n;i++) scanf("%d",&seq[i]);
	s.insert(0);s.insert(n);ss.insert(sum[n]);
	set<int>::iterator iter;
	multiset<LL>::iterator it;
	int l,r;
	for (i=1;i<=n;i++)
	{
		iter=s.lower_bound(seq[i]);
		r=*iter;l=*(--iter);
		ss.erase(ss.find(sum[r]-sum[l]));
		s.erase(l);s.erase(r);
		if (seq[i]!=r)
		{
			s.insert(seq[i]);
			s.insert(r);
			ss.insert(sum[r]-sum[seq[i]]);
		}
		if (seq[i]!=l+1)
		{
			s.insert(l);
			s.insert(seq[i]-1);
			ss.insert(sum[seq[i]-1]-sum[l]);
		}
		if (ss.size()>0)
		{
			it=ss.end();
			printf("%I64d\n",*(--it));
		}
		else
			printf("0\n");
	}
	return 0;
}

在网上看到一个绝妙的dsu的做法

我们反过来考虑这个过程，也就是往一个空序列里面不断加点

每加一个点，如果它的左边或右边已经有点，就合并，否则就单独一个团

只有更新过的团才可能更新答案，所以更新答案是O(1)的

记得开long long

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <utility>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <queue>
#include <deque>
#define x first
#define y second
#define mp make_pair
#define pb push_back
#define LL long long
#define Pair pair<int,int>
#define LOWBIT(x) x & (-x)
using namespace std;

const int MOD=1e9+7;
const int INF=0x7ffffff;
const int magic=348;

int n;
int a[100048];
int seq[100048];
LL ans[100048];
bool exist[100048];

int pre[100048];
LL sum[100048];
inline int find_anc(int x)
{
	if (x!=pre[x]) pre[x]=find_anc(pre[x]);
	return pre[x];
}
inline void update(int x,int y)
{
	x=find_anc(x);y=find_anc(y);
	pre[x]=y;
	sum[y]+=sum[x];
}

int main ()
{
	int i;
	scanf("%d",&n);
	for (i=1;i<=n;i++) scanf("%d",&a[i]);
	for (i=1;i<=n;i++) scanf("%d",&seq[n-i+1]);
	for (i=1;i<=n;i++) pre[i]=i;
	ans[n]=0;
	LL maxn=-1;
	for (i=1;i<=n-1;i++)
	{
		exist[seq[i]]=true;
		sum[seq[i]]=a[seq[i]];
		if (seq[i]==1)
		{
			if (exist[2]) update(1,2);
			maxn=max(maxn,sum[find_anc(1)]);
			ans[n-i]=maxn;
			continue;
		}
		if (seq[i]==n)
		{
			if (exist[n-1]) update(n-1,n);
			maxn=max(maxn,sum[find_anc(n)]);
			ans[n-i]=maxn;
			continue;
		}
		if (exist[seq[i]-1]) update(seq[i]-1,seq[i]);
		if (exist[seq[i]+1]) update(seq[i]+1,seq[i]);
		maxn=max(maxn,sum[find_anc(seq[i])]);
		ans[n-i]=maxn;
	}
	for (i=1;i<=n;i++) printf("%I64d\n",ans[i]);
	return 0;
}