收获:对于一个长度为k的区间是否存在
[
L
,
L
+
k
−
1
]
[L,L+k-1]
[L,L+k−1]的所有数字,可以用区间和和区间异或和来判别。
思路:对于前后残缺的区间可以预处理出前后缀中不同数字的个数,然后枚举分割点,对于完整的区间用区间和和区间异或和判断。
ll a[N], XOR[N], SUM[N];
ll n, k, sum, Xor;
int cot1[N], cot2[N];
void x_x()
{
unordered_map<ll, int> mp;
cot1[0] = cot2[n + 1] = 0;
f(i, 1, n)
{
if (mp[a[i]] == 0)cot1[i] = cot1[i - 1] + 1;
else cot1[i] = cot1[i - 1];
mp[a[i]]++;
}
mp.clear();
ff(i, n, 1)
{
if (mp[a[i]] == 0)cot2[i] = cot2[i + 1] + 1;
else cot2[i] = cot2[i + 1];
mp[a[i]]++;
}
}
bool check(int idx)
{
int i;
for (i = idx;i+k-1<= n;i += k)
{
ll nowsum = SUM[i + k - 1] - SUM[i - 1];
ll nowxor = XOR[i + k - 1] ^ XOR[i - 1];
if (nowsum != sum)return false;
if (nowxor != Xor)return false;
}
if (cot2[i]<n-i+1)return false;
return true;
}
int main()
{
//freopen("in.txt", "r", stdin);
int t;cin >> t;
while (t--)
{
scanf("%lld%lld", &n, &k);
f(i, 1, n)scanf("%lld", &a[i]);
x_x();
sum = 0, Xor = 0;
if (n >k)
{
f(i, 1, n)SUM[i] = SUM[i - 1] + a[i], XOR[i] = XOR[i - 1] ^ a[i];
f(i, 1, k)sum += i, Xor ^= i;
f(i, 0, k-1)
{
if (cot1[i]==i&& check(i + 1)) { puts("YES");goto fin; }
}
puts("NO");
}
else
{
f(i, 1, n - 1)
{
if (cot1[i]==i&&cot2[i+1]==n-i) { puts("YES");goto fin; }
}
puts("NO");
}
fin:;
}
return 0;
}