ABC-Prefix Equality-(思维转化+树状数组维护最值)

E

题意：
就是给你两个数组va和vb，现在又给你m个查询，每次问你va数组的前a个数组成的set和vb数组前b个数组成的set是否相同。

思考：
刚开始我以为是开个map第一位是位置，第二位就是set，然后空间复杂度是n*n的，但是题目给的1024MB，以为能冲过去。实际上对于set放到map里面时间也是比较大的，所以没那么简单。然后我就想，让前a和和前b个是否相同，可以直接在va数组每个下标维护这个数在vb中出现的最小位置。这样直接查询1到a的最大值是否<=b就行了，这样就能满足前面a个所有数都在vb数组的前b个里面。同理vb数组也要维护一个。然后查询的时候满足两个条件那就是yes。实际上和之前做的这个题：选数异或。本质思路是一样的。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
		
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 1e6+10,M = 2010;

int T,n,m,k;
int va[N];
int vb[N];
int idx[N],idx1[N];
int tr[N],tr1[N],R = 3e5+5;

map<int,int > mp,mp1;

int bit(int x)
{
	return x&(-x);
}

void update(int x,int val)
{
	while(x<=R)
	{
		tr[x] = max(tr[x],val);
		x += bit(x);
	}
}

int query(int x)
{
	int maxn = 0;
	while(x)
	{
		maxn = max(maxn,tr[x]);
		x -= bit(x);
	}
	return maxn;	
}

void update1(int x,int val)
{
	while(x<=R)
	{
		tr1[x] = max(tr1[x],val);
		x += bit(x);
	}
}

int query1(int x)
{
	int maxn = 0;
	while(x)
	{
		maxn = max(maxn,tr1[x]);
		x -= bit(x);
	}
	return maxn;	
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<=n;i++) cin>>vb[i];
	for(int i=n;i>=1;i--) mp[vb[i]] = i;
	for(int i=n;i>=1;i--) mp1[va[i]] = i;
	for(int i=1;i<=n;i++)
	{
		if(mp[va[i]]!=0) idx[i] = mp[va[i]];
		else idx[i] = inf;
		if(mp1[vb[i]]!=0) idx1[i] = mp1[vb[i]];
		else idx1[i] = inf;
	}
	for(int i=1;i<=n;i++)
	{
		update(i,idx[i]);
		update1(i,idx1[i]);
	}
	cin>>m;
	while(m--)
	{
		int a,b;
		cin>>a>>b;
		if(query(a)<=b&&query1(b)<=a) cout<<"Yes\n";
		else cout<<"No\n";
	}
	return 0;
}

总结：
多多思考。