結果

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

ソースコード

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;
	if(N == 1 && K == 1){
		cout << "Yes" << endl;
		cout << 1 << endl;
		return 0;
	}
	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 / 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