codeforeces 1187 D 单点修改区间最值用树状数组

网上的好多树状数组的区间最值都是错的啊。。。

修修补补过了这题不知道还有没有别的问题。。。

先当板子用这吧。。。以后碰到单点修改区间查询再测测

#include<bits/stdc++.h>
#define maxl 300010
 
using namespace std;
 
int n,ans;
int a[maxl],b[maxl],c[maxl],cnt[maxl],len[maxl];
vector <int> pos[maxl];
int h[maxl]; 

inline void prework()
{
	ans=1;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),cnt[i]=0,len[i]=0;
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]),pos[i].clear(),h[i]=0;
	for(int i=1;i<=n;i++)
		pos[a[i]].push_back(i),len[b[i]]++;
	for(int i=1;i<=n;i++)
	if(len[i]!=(int)(pos[i].size()))
	{
		ans=0;
		return;		
	}
}

inline void upd(int i,int x)
{
	int t;
	while(i<=n && i)
	{
		h[i]=c[i];  //网上代码这里都写=x???显然是错的啊 
		for(int k=1;k<(i&-i);k<<=1)
			h[i]=min(h[i],h[i-k]);
		i+=i&-i;
	}
}
 
inline int qry(int l,int r)
{
	int ret=n+1,len=r-l+1;
	while(len && r)
	{
		if(len<(r&-r))
		{
			ret=min(ret,c[r]);
			r--;len--;
		}else
		{
			ret=min(ret,h[r]);
			len-=(r&-r);
			r-=r&-r;				
		}
	}
	return ret;
}
 
inline void mainwork()
{
	if(!ans) return;
	for(int i=1;i<=n;i++)
	{
		if(pos[i].size()>0)
			c[i]=pos[i][0];
		else
			c[i]=n+1;
		upd(i,c[i]);
	}
	int id;
	for(int i=1;i<=n;i++)
	{
		cnt[b[i]]++;
		id=pos[b[i]][cnt[b[i]]-1];
		if(qry(1,b[i])<id)
		{
			ans=0;
			return;
		}
		if(cnt[b[i]]==len[b[i]])
			c[b[i]]=n+1;
		else
			c[b[i]]=pos[b[i]][cnt[b[i]]];
		upd(b[i],c[b[i]]);
	}
}
 
inline void print()
{
	if(ans)
		puts("YES");
	else
		puts("NO");
}
 
int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}