Odd Sum Segments CodeForces - 1196B

题目传送门
分析:题目意思是给你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了好几次…

发布了40 篇原创文章 · 获赞 2 · 访问量 860

猜你喜欢

转载自blog.csdn.net/qq_43851311/article/details/103169697