Codeforces Round #640 (Div. 4) B. Same Parity Summands（思維）

題目點這裏

題意

給你一個數n要求分成k份且每份奇偶性相同，可以實現則輸出YES和其中一種可能，否則輸出NO。

思路

我們可以分成兩大類考慮：分成k份由奇數構成和k份偶數構成。
1.奇數構成的情況：
先將k個桶裏都放入一個1，則剩下n-k個數。若n-k爲偶數，則把n-k都放入其中一個桶裏，構成數組 1 1 1 1 …n-k+1 滿足題目要求。若n-k爲奇數，因爲奇數只能拆成若干個偶數+奇數個奇數的形式，所以必然有幾個桶裏的數字由於這幾個奇數變成偶數，不滿足都是奇數的情況，輸出NO。
2.偶數構成的情況：
類比奇數，k個桶每個放入一個2，判斷n-2k的奇偶性，偶數則可行，輸出
2 2 2 2 2 … n-2k+2 否則NO。

AC代碼

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e5+10;

int main()
{
    int n,k,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&k);
        if(n>=k&&(n-k)%2==0)
        {
            printf("YES\n");
            for(int i=0; i<k-1; i++)
                printf("%d ",1);
            printf("%d\n",n-k+1);
            continue;//這裏的continue觀摩某大佬學到的，省去了flag變量，很巧妙。
        }
        else if(n-2*k>=0&&(n-2*k)%2==0)
        {
            printf("YES\n");
            for(int i=0; i<k-1; i++)
                printf("%d ",2);
            printf("%d\n",n-2*k+2);
           continue;
        }
        printf("NO\n");
    }
    return 0;
}