hihocoder Offer收割 编程练习赛108 T4 01匹配 题解

题目传送门

题目大意： 给出一个长度为 $n$ 的序列，如果存在一对 $x,y$ 满足 $x<y$ 且 $a_x=1,a_y=0$，那么我们称 $x,y$ 匹配成功，现在多次询问，每次给出 $l,r$，询问区间 $[l,r]$ 中的最大匹配数。

题解

发现这样一个性质：假如我们先在全局中找出每个 $x$ 所对应的 $y$，并且对于每个 $x$，找尽可能近的 $y$，那么对于每一次询问的区间 $[l,r]$ 中的值为 $1$ 的位置，假如它对应的那个位置不在区间内，那么它就一定不能成功匹配。

TLE前的想法：那么设 $f[i]$ 表示与 $i$ 对应的位置，然后用分块去维护这个 $f$ 即可。

……

考虑离线去做这个东西，将询问按左端点从大到小排，然后用线段树去维护。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 300010

int n,m;
int a[maxn],f[maxn],zhan[maxn],t=0;
struct node{
	int l,r,sum;
	node *zuo,*you;
	node(int x,int y)
	{
		l=x,r=y;sum=0;
		if(l<r)
		{
			int mid=l+r>>1;
			zuo=new node(l,mid);
			you=new node(mid+1,r);
		}
	}
	void change(int x)
	{
		sum++;
		if(l==r)return;
		int mid=l+r>>1;
		if(x<=mid)zuo->change(x);
		else you->change(x);
	}
	int ask(int x,int y)
	{
		if(l==x&&r==y)return sum;
		int mid=l+r>>1;
		if(y<=mid)return zuo->ask(x,y);
		else if(x>=mid+1)return you->ask(x,y);
		else return zuo->ask(x,mid)+you->ask(mid+1,y);
	}
};
node *root=NULL;
void work()
{
	for(int i=n;i>=1;i--)
	if(a[i]==0)zhan[++t]=i;
	else f[i]=zhan[t--];
	root=new node(1,n);
}
struct que{int x,y,pos;};
que ask[maxn];
bool cmp(que x,que y){return x.x>y.x;}
int ans[maxn];

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	work();
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	scanf("%d %d",&ask[i].x,&ask[i].y),ask[i].pos=i;
	sort(ask+1,ask+m+1,cmp);
	int now=1;
	for(int i=n;i>=1;i--)
	{
		if(f[i]!=0)root->change(f[i]);
		while(now<=m&&ask[now].x==i)ans[ask[now].pos]=root->ask(ask[now].x,ask[now].y),now++;
	}
	for(int i=1;i<=m;i++)
	printf("%d\n",ans[i]);
}