CF1746A题解

题目链接

题意

题目描述

给定一个长度为 $n$ 的只包含 $0$ 和 $1$ 的序列 $a$ 和一个数字 $k$，可以进行如下的操作各若干次：

• 选定一个 $i\in [1,n-1]$，将 $a_i$ 和 $a_{i+1}$ 这两个数替换为它们的最小值。进行此操作后，序列的长度 $n$ 减少 $1$。

• 选定一个 $i\in [1,n-k+1]$，将 $a_i$ 到 $a_{i+k-1}$ 这 $k$ 个数替换为它们的最大值。进行此操作后，序列的长度 $n$ 减少 $k-1$。

判断经过若干次操作后是否能使 $n=1$ 且 $a_1=1$。

输入

$t$ 组数据，每组数据第一行输入两个整数 $n$ 和 $k$，第二行输入 $n$ 个整数 $a_i$。

输出

若可以，输出 YES，否则输出 NO。

思路

可以证明只要序列中里有 $1$ 就可以把序列变成单独一个 $1$。

若序列中有 $1$，输出 YES，否则输出 NO。

代码

#include<bits/stdc++.h>
using namespace std;
int t,n,k,f,a;
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>n>>k;
		f=0;
		for(int i=1;i<=n;i++)
		{
			cin>>a;
			if(a)
				f=1;
		}
		cout<<(f?"YES\n":"NO\n");
	}
	return 0;
}