Week13 A - TT 的神秘任务1（必做）

Week13 A - TT 的神秘任务1（必做）
这一天，TT 遇到了一个神秘人。

神秘人给了两个数字，分别表示 n 和 k，并要求 TT 给出 k 个奇偶性相同的正整数，使得其和等于 n。

例如 n = 10，k = 3，答案可以为 [4 2 4]。

TT 觉得这个任务太简单了，不愿意做，你能帮他完成吗？

本题是SPJ

Input

第一行一个整数 T，表示数据组数，不超过 1000。

之后 T 行，每一行给出两个正整数，分别表示 n（1 ≤ n ≤ 1e9）、k（1 ≤ k ≤ 100）。

Output
如果存在这样 k 个数字，则第一行输出 “YES”，第二行输出 k 个数字。

如果不存在，则输出 “NO”。

Example
Input

8
10 3
100 4
8 7
97 2
8 8
3 10
5 3
1000000000 9

Output

YES
4 2 4
YES
55 5 5 35
NO
NO
YES
1 1 1 1 1 1 1 1
NO
YES
3 1 1
YES
111111110 111111110 111111110 111111110 111111110 111111110 111111110 111111110 111111120

解题思路
有奇偶两种情况
首先平均分成k份
在随机加上1
Code

#include<iostream>
#include<string.h>
using namespace std;
int T;
int n,k;
int s[10000];
int nn(int n,int k){
    memset(s,0,sizeof(s));
    int y=n/k;
	for(int i=0;i<k;i++){
		s[i]=y;
	}
	while(1){
			int sum=0;
			for(int i=0;i<k;i++){
				//cout<<s[i]<<endl;
				sum+=s[i];
			}//cout<<endl;
			if(sum>n){
				return 0;
			}
			if(sum==n){
				cout<<"YES"<<endl;
				for(int i=0;i<k;i++){
					if(i!=0){
						cout<<' ';
					}cout<<s[i];
				}cout<<endl;
				return 1;
			}
			s[0]+=2;
		}

}
int nnn(int n,int k){
    memset(s,0,sizeof(s));
    int y=n/k;
	if(y==1)y++;
	else y--;
	for(int i=0;i<k;i++){
		s[i]=y;
	}
	while(1){
			int sum=0;
			for(int i=0;i<k;i++){
				//cout<<s[i]<<endl;
				sum+=s[i];
			}//cout<<endl;
			if(sum>n){
				return 0;
			}
			if(sum==n){
				cout<<"YES"<<endl;
				for(int i=0;i<k;i++){
					if(i!=0){
						cout<<' ';
					}cout<<s[i];
				}cout<<endl;
				return 1;
			}
			s[0]+=2;
		}

}
int main(){
	cin>>T;
	for(int i=0;i<T;i++){
		cin>>n>>k;
		if(k>n){cout<<"NO"<<endl;
		}else{
			if(!nn(n,k)){
					if(!nnn(n,k)){
						cout<<"NO"<<endl;
					}	
				}
		}
		
	}
}