hdu6378 度度熊玩数组

链接

http://acm.hdu.edu.cn/showproblem.php?pid=6378

题解

离线，倒着做往里加数
每次新出现的区间就是所有包含这个点的区间
如果我对每个长度较短的区间里的元素做某种$O(f(n))$的操作，那么总的操作数是$O(nlo{g}_2(n)f(n))$，这个的原理类似于启发式合并
那么我肯定是枚举短区间里的元素，然后在长区间里查询绝对值和某个数最接近的数
最显然的做法是主席树，不过$hdu$的空间限制有点堪忧…然后可以发现这个东西是联通块的合并，不妨再每个联通快上维护一个$set$，然后启发式合并
总的复杂度$O(nlo{g}_2^{2}n)$

代码

#include<bits/stdc++.h>
#define maxn 100010
#define linf (1ll<<60)
#define iinf 0x7fffffff
#define eps 1e-8
#define cl(x) memset(x,0,sizeof(x))
#define mod 998244353ll
using namespace std;
typedef long long ll;
int read(int x=0)
{
	int c, f=1;
	for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
	for(;isdigit(c);c=getchar())x=x*10+c-48;
	return f*x;
}
struct UnionFind
{
	int f[maxn], sz[maxn];
	set<int> st[maxn];
	void init(int n)
	{
		for(auto i=1;i<=n;i++)f[i]=i, st[i].clear(), sz[i]=1;
	}
	int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
	void merge(int x, int y)
	{
		auto fx=find(x), fy=find(y);
		if(st[fx].size()>st[fy].size())swap(fx,fy);
		for(auto x:st[fx])st[fy].insert(x);
		st[fx].clear();
		f[fx]=find(fy);
        sz[fy]+=sz[fx];
	}
}uf;
int getmin(set<int> &s, int x)
{
	if(s.find(x)!=s.end())return 0;
	int ans(iinf);
	s.emplace(x);
	auto it=s.find(x);
	if(it!=s.begin())
	{
		it--;
		ans=min(ans,abs(*it-x));
		it++;
	}
	it++;
	if(it!=s.end())ans=min(ans,abs(*it-x));
	it--;
	s.erase(it);
	return ans;
}
int a[maxn], s[maxn], mark[maxn], lis[maxn], ans[maxn];
int main()
{
	int n, K, i, j, k, l, r;
	while(~scanf("%d%d",&n,&K))
	{
		for(i=1;i<=n;i++)a[i]=read(), s[i]=s[i-1]+a[i];
		for(i=1;i<=n;i++)lis[i]=read();
		uf.init(n);
		for(i=1;i<=n;i++)uf.st[i].insert(s[i]);
		cl(mark);
		for(i=n;i;i--)
		{
			auto p=lis[i], l(1), r(1);
			ans[i]=iinf;
			mark[p]=1;
			if(mark[p-1])l+=uf.sz[uf.find(p-1)];
			if(mark[p+1])r+=uf.sz[uf.find(p+1)];
			if(l==1 and r==1){ans[i]=abs(a[p]-K);continue;}
			if(l<r)
			{
				uf.merge(p,p+1);
				for(j=p-l;j<p;j++)
					ans[i]=min( ans[i], getmin(uf.st[uf.find(p)],s[j]+K) );
				if(l>1)uf.merge(p-1,p);
			}
			else
			{
				for(j=p;j<=p+r-1;j++)
				{
					ans[i]=min( ans[i], getmin(uf.st[uf.find(p-1)],s[j]-K) );
					ans[i]=min( ans[i], abs(s[j]-s[p-l]-K) );
				}
				uf.merge(p,p-1);
				if(r>1)uf.merge(p,p+1);
			}
		}
		for(i=n-1;i;i--)ans[i]=min(ans[i],ans[i+1]);
		for(i=1;i<=n;i++)printf("%d\n",ans[i]);
	}
	return 0;
}