結果

問題 No.942 プレゼント配り
ユーザー chocopuu
提出日時 2019-12-05 04:08:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,668 bytes
コンパイル時間 2,067 ms
コンパイル使用メモリ 173,936 KB
実行使用メモリ 7,880 KB
最終ジャッジ日時 2024-12-15 18:49:13
合計ジャッジ時間 3,873 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 17 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define REP(i,n) for(int i = 0;i < (int)(n);i++)
#define RREP(i,n) for(int i = (int)n-1;i >= 0;i--)
#define FOR(i,s,n) for(int i = s;i < (int)n;i++)
#define RFOR(i,s,n) for(int i = (int)n-1;i >= s;i--)
#define ALL(a) a.begin(),a.end()
#define IN(a, x, b) (a<=x && x<b)
template<class T>inline bool CHMAX(T&a,T b){if(a<b){a = b;return true;}return false;}
template<class T>inline bool CHMIN(T&a,T b){if(a>b){a = b;return true;}return false;}
constexpr long long INF = 1e18;

signed main(){
	int N,K;
	cin>>N>>K;
	int sum = N * (N + 1) / 2;
	if(sum % K){
		cout << "No" << endl;
		return 0;
	}
	vector<vector<int>>ans(K);
	if(K % 2){
		if(N == K){
			cout << "No" << endl;
			return 0;
		}else if(N == 2 * K){
			REP(i,K){
				ans[i].push_back(i);
				ans[i].push_back(2*K-i-1);
			}
		}else if((N / K) % 2 == 0){
			int turn = N / K;
			REP(i,turn / 2){
				REP(j,K){
					ans[j].push_back(i * 2 * K + j);
					ans[j].push_back(i * 2 * K + 2 * K - j - 1);
				}
			}
		}else{
			REP(i,2*K){
				if(i % 2)ans[i % K].push_back(K + i / 2);
				else ans[i % K].push_back(i / 2);
			}
			REP(i,K){
				ans[i].push_back(3 * K - i - 1);
			}
			int turn = N / K - 3;
			REP(i,turn / 2){
				REP(j,K){
					ans[j].push_back(i * 2 * K + 3 * K + j);
					ans[j].push_back(i * 2 * K + 3 * K + 2 * K - j - 1);
				}
			}
		}
	}else{
		int turn = N / K;
		REP(i,turn / 2){
			REP(j,K){
				ans[j].push_back(i * 2 * K + j);
				ans[j].push_back(i * 2 * K + 2 * K - j - 1);
			}
		}
	}
	cout << "Yes" << endl;
	REP(i,K){
		REP(j,ans[i].size()){
			if(j)cout << " ";
			cout << ans[i][j] + 1;
		}
		cout << endl;
	}
}
0