题目
题意:
给两个数组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 1≤n≤3⋅105,1≤ai≤n,1≤bi≤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;
}