传送门:https://codeforces.com/problemset/problem/1187/D
比赛中写了个假的FST了,不过还好E题是个水题而且写得够快,还是上了一大波分
那天晚上网友给了个hack数据
5
1 3 5 4 2
1 2 3 5 4
这个是YES,然后我就发现我没了,然后去hack了十多个人,结果发现edu赛后hack不加分。。。
后来补题瞄了一眼题解,发现把整体排序等价为两两交换后,关掉题解我瞎写了一通,然后WA再第20个点
5
1 3 4 5 2
1 2 4 5 3
这个是NO,然后通过这两个数据就能想到正解了。
对于线段树的每个下标i,我们存i这个数字下次在a数组中出现的位置。
非法情况就是当我要放一个数字i的时候,当前这个 i 是b中的第cnt[i]个i在a数组里应该放在idx=pos[i][cnt[i]-1],这个时候查看1-i中下一个要放的位置有没有小于idx的,如果有,那么就非法,输出NO。
上述例子的流程就是按b的顺序先放1,然后把1位置更新为n+1(后面已经没有1了),然后更新2为n+1,枚举到4的时候,发现1-3中1,2都为n+1,但是3为2,说明当前在a数组2号位置在b数组中实在4后面的,而3不能跟4交换,让3到4后面,所以非法
#include<bits/stdc++.h>
#define maxl 300010
using namespace std;
int n,ans;
int a[maxl],b[maxl],cnt[maxl],len[maxl];
vector <int> pos[maxl];
struct node
{
int l,r;
int mini;
int tag;
}tree[maxl*4];
inline void build(int k,int l,int r)
{
tree[k].l=l;tree[k].r=r;
if(l==r)
{
if(pos[l].size()==0)
tree[k].mini=n+1;
else
tree[k].mini=pos[l][0];
return;
}
int mid=(l+r)>>1;
build(k<<1,l,mid);build(k<<1|1,mid+1,r);
tree[k].mini=min(tree[k<<1].mini,tree[k<<1|1].mini);
}
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();
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;
}
build(1,1,n);
}
inline int qry(int k,int l,int r)
{
if(tree[k].l==l && tree[k].r==r)
return tree[k].mini;
int mid=(tree[k].l+tree[k].r)>>1;
int ret;
if(r<=mid)
ret=qry(k<<1,l,r);
else if(l>mid)
ret=qry(k<<1|1,l,r);
else
ret=min(qry(k<<1,l,mid),qry(k<<1|1,mid+1,r));
return ret;
}
inline void upd(int k,int l,int x)
{
int mid=(tree[k].l+tree[k].r)>>1;
if(tree[k].l==tree[k].r)
{
tree[k].mini=x;
return;
}
if(l<=mid)
upd(k<<1,l,x);
else
upd(k<<1|1,l,x);
tree[k].mini=min(tree[k<<1].mini,tree[k<<1|1].mini);
}
inline void mainwork()
{
if(!ans) return;
int id;
for(int i=1;i<=n;i++)
{
cnt[b[i]]++;
id=pos[b[i]][cnt[b[i]]-1];
if(qry(1,1,b[i])<id)
{
ans=0;
return;
}
if(cnt[b[i]]==len[b[i]])
upd(1,b[i],n+1);
else
upd(1,b[i],pos[b[i]][cnt[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;
}