結果

問題 No.942 プレゼント配り
ユーザー ianCK
提出日時 2019-12-09 08:14:20
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 21 ms / 2,000 ms
コード長 988 bytes
コンパイル時間 390 ms
コンパイル使用メモリ 39,016 KB
実行使用メモリ 8,596 KB
最終ジャッジ日時 2024-06-11 19:36:20
合計ジャッジ時間 1,926 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:10:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   10 |         scanf("%d%d", &n, &k);
      |         ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <stdio.h>
#include <vector>
using namespace std;
constexpr int kN = int(2E5 + 10);
#define PB push_back
vector<int> ans[kN];

int main() {
	int n, k, need, now;
	scanf("%d%d", &n, &k);
	if (n == 1 && k == 1) {
		printf("Yes\n1\n");
		return 0;
	}
	if (n == k || (long long int) n * (long long int) (n + 1) % (k << 1) != 0) printf("No\n");
	else {
		need = n / k;
		now = 1;
		if (need & 1) {
			for (int i = 1; i <= k; i++) ans[i].PB(i);
			for (int i = 1, j = 0; i <= k; i++, j += 2) {
				if (j >= k) j -= k;
				ans[i].PB(k * 3 - j);
				ans[i].PB((k * 3 + 3) / 2 + j - i);
			}
			need -= 3;
			now = k * 3 + 1;
		}
		while (need) {
			for (int i = 1; i <= k; i++) ans[i].PB(now++);
			for (int i = k; i >= 1; i--) ans[i].PB(now++);
			need -= 2;
		}
		need = n / k;
		printf("Yes\n");
		for (int i = 1; i <= k; i++) {
			printf("%d", ans[i][0]);
			for (int j = 1; j < need; j++) printf(" %d", ans[i][j]);
			printf("\n");
		}
	}
}
0