题意
题目描述
给定一个长度为 n n n 的只包含 0 0 0 和 1 1 1 的序列 a a a 和一个数字 k k k,可以进行如下的操作各若干次:
-
选定一个 i ∈ [ 1 , n − 1 ] i\in[1,n-1] i∈[1,n−1],将 a i a_i ai 和 a i + 1 a_{i+1} ai+1 这两个数替换为它们的最小值。进行此操作后,序列的长度 n n n 减少 1 1 1。
-
选定一个 i ∈ [ 1 , n − k + 1 ] i\in[1,n-k+1] i∈[1,n−k+1],将 a i a_i ai 到 a i + k − 1 a_{i+k-1} ai+k−1 这 k k k 个数替换为它们的最大值。进行此操作后,序列的长度 n n n 减少 k − 1 k-1 k−1。
判断经过若干次操作后是否能使 n = 1 n=1 n=1 且 a 1 = 1 a_1=1 a1=1。
输入
t t t 组数据,每组数据第一行输入两个整数 n n n 和 k k k,第二行输入 n n n 个整数 a i a_i ai。
输出
若可以,输出 YES
,否则输出 NO
。
思路
可以证明只要序列中里有 1 1 1 就可以把序列变成单独一个 1 1 1。
若序列中有
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;
}