题意:给定数列长度n,以及子序列数k。问:是否能将数列分为k个序列,且每个序列和都为奇数。
对于每个区间而言,偶数的贡献为零,偶数的多少并不会改变序列和的奇偶性,因此,只需要把握奇数的个数即可。若证每个序列和为奇数,即证每个序列共有奇数个奇数。从分配的角度上看,只需令前k-1个序列有1个奇数,第k个区间有序列总奇数个数-k个奇数个奇数。所以,只要保证第k个区间有奇数个奇数,且总奇数数大于等于k即可。
代码如下:
#include<bits/stdc++.h>
const int MAXN = 2e5+10;
int find_n[MAXN];
using namespace std;
void solve()
{
int find_i = 1;
int num1 = 0;
int n,k;
cin>>n>>k;
vector<int> a;
for(int i = 0; i < n; i++){
int x;
cin>>x;
if(x %2 != 0) {
num1++;
if(find_i < k ){
find_n[find_i++] = i+1;
}
}
}
find_n[find_i] = n;
if(((num1 - k)% 2 == 0) && num1 >= k){
cout<<"YES"<<'\n';
for(int i = 1 ;i <= find_i; i++) cout<<find_n[i]<<" ";
cout<<"\n";
}
else cout<<"NO"<<"\n";
}
int main()
{
int T;
cin>>T;
while(T--){
solve();
}
}