D. Subarray Sorting(区间极小+分析维护)

题目

题意:

    给两个数组a,b。你可以使a的任意区间从小到大排序,判断是否可以通过操作使得a数组变为b数组。
     1 n 3 1 0 5 , 1 a i n , 1 b i n 1≤n≤3⋅10^5,1≤a_i≤n,1≤b_i≤n

分析:

    我们考虑从左到右依次满足a[i]==b[i],对于一个b[i]来说,肯定是要在a数组中找到相应的值所在的位置x,若这个值[1,x]这个区间的最小值,那么才可以移动过来,否则比它小的元素必然会阻碍他。为什么是[1,x]呢?因为之前匹配的元素放到了前面去,那么那些未匹配的元素实际上移动了,但是我们并没有维护这个移动的过程,即所有未匹配过在[1,x]这一区间的值的位置都在b[i]之后了。对于已经匹配好的元素,即把他们放到了最前面去了,所以就可以不必考虑他们的存在,设为最大值即可。区间最小用线段树维护一下即可。

#include <iostream>
#include <queue>
using namespace std;

queue<int> q[300005];
int b[300005];

struct node{
	int l,r,minx;
}a[300005*4];

void update(int x)
{
	a[x].minx = min(a[2*x].minx,a[2*x+1].minx); 
}

void build(int x,int l,int r)
{
	a[x].l = l;
	a[x].r = r;
	if( l == r )
	{
		a[x].minx = 0;
		return;
	}
	int mid = ( l + r ) / 2;
	build(2*x,l,mid);
	build(2*x+1,mid+1,r);
	update(x);
}

void change(int x,int l,int r,int k)
{
	if( a[x].l == l && a[x].r == r )
	{
		a[x].minx += k;
		return;
	}
	int mid = ( a[x].l + a[x].r ) / 2;
	if( l > mid ) change(2*x+1,l,r,k);
	else if( r <= mid ) change(2*x,l,r,k);
	else
	{
		change(2*x,l,mid,k);
		change(2*x+1,mid+1,r,k);
	}
	update(x);
}

int query(int x,int l,int r)
{
	if( a[x].l == l && a[x].r == r ) return a[x].minx;
	int mid = ( a[x].l + a[x].r ) / 2;
	if( l > mid ) return query(2*x+1,l,r);
	else if( r <= mid ) return query(2*x,l,r);
	else return min(query(2*x,l,mid),query(2*x+1,mid+1,r));
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int t;
	cin >> t;
	while( t-- )
	{
		int n;
		cin >> n;
		build(1,1,n);
		for (int i = 1; i <= n; i++) 
		{
			while( !q[i].empty() ) q[i].pop();
		}
		for (int i = 1; i <= n; i++)
		{
			int x;
			cin >> x;
			q[x].push(i);
			change(1,i,i,x);
		}
		for (int i = 1; i <= n; i++)
		{
			cin >> b[i];
		}
		int flag = 0;
		for (int i = 1; i <= n; i++)
		{
			//cout << i << '\n';
			if( q[b[i]].empty() )
			{
				flag = 1;
				break;
			}
			int x = q[b[i]].front();
			q[b[i]].pop();
			int t = query(1,1,x);
			//cout << "asd " << x << ' ' <<t << '\n'; 
			if( t != b[i] )
			{
				flag = 1;
				break;
			}
			change(1,x,x,3e5);
		}
		if( flag ) cout << "NO" << '\n';
		else cout << "YES" << '\n';
	}
	return 0;
}

发布了132 篇原创文章 · 获赞 6 · 访问量 7927

猜你喜欢

转载自blog.csdn.net/weixin_44316314/article/details/104868956