题目传送门
分析:题目意思是给你n个数,判断能否分为k份且每份的和为奇数;
如果不能分 output NO;
能分的话 output YES , 并且输出每个区间右端
分法为 [1,r1] , [r1+1;r2] , [r2+1,r3] ,…, [rk−1+1,n],需要注意的是最后一个一定是n;
讨论如何分,奇数相加为偶数,所以我需要保证每个区间最少一个奇数,
就让每个区间只有一个奇数,那么剩下的奇数应该有偶数个;
满足奇数的个数ans>=k&&((ans-k)%2==0)时,只需要找到k-1个奇数并输出位置,最后输出n 即可;
代码如下:
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=2e5+100;
int main()
{
ll q;
scanf("%lld",&q);
while(q--)
{
ll n,k;
ll a[maxn];
ll ans=0;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
if(a[i]&1)
ans++;
}
if(ans%2==k%2&&ans>=k)
{
printf("YES\n");
k--;
for(int i=1;i<=n;i++)
{
if(!k)
{
printf("%lld",n);
break;
}
if(a[i]&1)
{
printf("%d ",i);
k--;
}
}
}
else
printf("NO");
if(q)
printf("\n");
}
return 0;
}
这道题思路不难,写起来也不难,但是在跳出循环哪里出了问题,wa了好几次…