2020暑期牛客第六场 K-Bag(思维,前缀)

收获:对于一个长度为k的区间是否存在 [ L , L + k − 1 ] [L,L+k-1] [L,L+k1]的所有数字,可以用区间和和区间异或和来判别。
思路:对于前后残缺的区间可以预处理出前后缀中不同数字的个数,然后枚举分割点,对于完整的区间用区间和和区间异或和判断。

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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值