题解：CF2211C1 Equal Multisets (Easy Version)

思路

考虑什么情况下不成立。
如果 $n>2\times k$，一定需要使得每个 $a_i$ 都与 $b_i$ 相等，否则一定会存在一个子序列包含某个不该在此子序列中的值而少了一个应该在此子序列中的值；而如果 $n\le 2\times k$，在中间的某一段由于被每个子序列都使用且没有被分开使用，所以可以是乱序的。
换句话说，我们发现对于每个 $k$，都有区间 $[1,n-k]$ 和区间 $[k,n]$ 被多个子序列使用，且某个值被一些区间使用而不被另一些区间使用，所以必须使得 $\sum_{i=1}^{n-k} a_i=b_i$ 且 $\sum_{i=k}^{n} a_i=b_i$。
而如果有某个值出现或被给出的次数超过了一次，则一定不成立，因为它无法同时出现在两个位置，这样会导致某个区间中的值少一个，无法与 $a$ 中的子序列相等。
按照上面的两步模拟即可。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include<bits/stdc++.h>
#define fir first
#define sec second 
#define int long long
#define pii pair<int,int>
#define fep(i,s,e) for(int i=s;i<e;i++)
#define pef(i,s,e) for(int i=s;i>e;i--)
#define rep(i,s,e) for(int i=s;i<=e;i++)
#define per(i,s,e) for(int i=s;i>=e;i--)
namespace FastIO{
	template<typename T>inline void read(T &x){
		x=0;int f=1;char c=getchar();
		for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
		for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+(c^48);x*=f;
	}
	template<typename T,typename...Args>
    inline void read(T &x,Args&...args){
        read(x);
        read(args...);
    }
	template<typename T>void print(T x){
		if(x<0)x=-x,putchar('-');
		if(x>9)print(x/10);
		putchar((x%10)^48);
	}
}
using namespace std;
using namespace FastIO;
const int N=2e5+47;
int T,n,k,a[N],b[N],vis[N];
signed main(){
	read(T);
	while(T--){
		read(n,k);
		int l=n-k+1,r=k,ans=1;
		rep(i,1,n)vis[i]=0;
		rep(i,1,n)read(a[i]);
		rep(i,1,n){
			read(b[i]);
			if(b[i]!=-1){
				vis[b[i]]++;
				if(vis[b[i]]>1)ans=0;
			}
		}
		rep(i,1,l-1){
			if(vis[a[i]]&&a[i]!=b[i])ans=0;
			if(b[i]!=-1&&a[i]!=b[i])ans=0;
		}
		rep(i,r+1,n){
			if(vis[a[i]]&&a[i]!=b[i])ans=0;
			if(b[i]!=-1&&a[i]!=b[i])ans=0;
		}
		if(ans==1)puts("YES");
		else puts("NO");
	}
}