[LOJ]#2310. 「APIO2017」斑斓之地 主席树

Solution

Update：
去洛谷上看吧链接。
我写的太垃圾了。

把有河流的格子称为黑点，其它称为白点，那么连通块数可以这样算：矩形内白点数$-1\times 2$白矩形数$-2\times 1$白矩形数$+2\times 2$白矩形数。
当且仅当白点包了一个环把黑点包住时会多减$1$。
由于黑点是连通的，只有这种情况要特判。判下黑点横纵坐标最大最小值是否被包在矩形中就可以了。
这个东西好像在冬令营上提到了一下，但是没有听清，路过的大佬如果知道可以告诉我一下。我只能够感性地理解一下。
然后这四个东西用主席树维护一下就好了。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=200010;
const int inf=2147483647;
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
unordered_map<int,int>mp[Maxn];
int R,C,m,q,sr,sc,px[Maxn],py[Maxn],tot=0,mxx,mxy,mnx,mny;
char str[Maxn];
struct Seg
{
	int lc[Maxn*20],rc[Maxn*20],c[Maxn*20],root[Maxn],tot;
	void insert(int &u,int l,int r,int p)
	{
		if(!u)u=++tot;
		c[u]++;if(l==r)return;
		int mid=l+r>>1;
		if(p<=mid)insert(lc[u],l,mid,p);
		else insert(rc[u],mid+1,r,p);
	}
	void merge(int &u1,int u2)
	{
		if(!u1){u1=u2;return;}
		if(!u2)return;
		c[u1]+=c[u2];
		merge(lc[u1],lc[u2]),merge(rc[u1],rc[u2]);
	}
	int query(int L,int R,int l,int r,int fl,int fr)
	{
		if(!R)return 0;
		if(fl==l&&fr==r)return c[R]-c[L];
		int mid=l+r>>1;
		if(fr<=mid)return query(lc[L],lc[R],l,mid,fl,fr);
		if(fl>mid)return query(rc[L],rc[R],mid+1,r,fl,fr);
		return query(lc[L],lc[R],l,mid,fl,mid)+query(rc[L],rc[R],mid+1,r,mid+1,fr);
	}
}T1,T2,T3,T4;
vector<int>h1[Maxn],h2[Maxn],h3[Maxn],h4[Maxn];
int main()
{
	R=read(),C=read(),m=read(),q=read();
	sr=read(),sc=read();
	mxx=mnx=sr,mxy=mny=sc;
	if(m)scanf("%s",str+1);
	for(int i=1;i<=m;i++)
	{
		px[++tot]=sr,py[tot]=sc;mp[sr][sc]=1;
		char ch=str[i];
		if(ch=='N')sr--;
		else if(ch=='S')sr++;
		else if(ch=='W')sc--;
		else sc++;
		mxx=max(mxx,sr),mnx=min(mnx,sr);
		mxy=max(mxy,sc),mny=min(mny,sc);
	}
	px[++tot]=sr,py[tot]=sc;mp[sr][sc]=1;
	T1.tot=T2.tot=T3.tot=T4.tot=0;
	T1.root[0]=T2.root[0]=T3.root[0]=T4.root[0]=0;
	for(int i=1;i<=tot;i++)
	{
		int x=px[i],y=py[i];
		if(mp[x][y]==2)continue;
		mp[x][y]=2;
		h1[x].push_back(y);
		if(y<C)h2[x].push_back(y);if(y>1&&!mp[x][y-1])h2[x].push_back(y-1);
		if(x<R)h3[x].push_back(y);if(x>1&&!mp[x-1][y])h3[x-1].push_back(y);
		if(x<R&&y<C)h4[x].push_back(y);
		if(x<R&&y>1&&!mp[x][y-1])h4[x].push_back(y-1);
		if(x>1&&y<C&&!mp[x-1][y]&&!mp[x-1][y+1])h4[x-1].push_back(y);
		if(x>1&&y>1&&!mp[x-1][y]&&!mp[x-1][y-1]&&!mp[x][y-1])h4[x-1].push_back(y-1);
	}
	for(int x=1;x<=R;x++)
	{
		for(int i=0;i<h1[x].size();i++){int y=h1[x][i];T1.insert(T1.root[x],1,C,y);}
		for(int i=0;i<h2[x].size();i++){int y=h2[x][i];T2.insert(T2.root[x],1,C,y);}
		for(int i=0;i<h3[x].size();i++){int y=h3[x][i];T3.insert(T3.root[x],1,C,y);}
		for(int i=0;i<h4[x].size();i++){int y=h4[x][i];T4.insert(T4.root[x],1,C,y);}
		T1.merge(T1.root[x],T1.root[x-1]);
		T2.merge(T2.root[x],T2.root[x-1]);
		T3.merge(T3.root[x],T3.root[x-1]);
		T4.merge(T4.root[x],T4.root[x-1]);
	}
	while(q--)
	{
		LL X1=read(),Y1=read(),X2=read(),Y2=read(),ans=0;
		if(X1<mnx&&X2>mxx&&Y1<mny&&Y2>mxy)ans++;
		LL c1=(X2-X1+1)*(Y2-Y1+1)-T1.query(T1.root[X1-1],T1.root[X2],1,C,Y1,Y2);
		LL c2=(X2-X1+1)*(Y2-Y1)-T2.query(T2.root[X1-1],T2.root[X2],1,C,Y1,Y2-1);
		LL c3=(X2-X1)*(Y2-Y1+1)-T3.query(T3.root[X1-1],T3.root[X2-1],1,C,Y1,Y2);
		LL c4=(X2-X1)*(Y2-Y1)-T4.query(T4.root[X1-1],T4.root[X2-1],1,C,Y1,Y2-1);
		ans+=(c1-c2-c3+c4);
		printf("%lld\n",ans);
	}
}