#405443 (C) No.942 プレゼント配り

• 問題: No.942 プレゼント配り
• ユーザー: bal4u
• 提出日時: 2019-12-06 06:17:03
• 言語: C (gcc 12.3.0)
• 結果: WA
• 実行時間: -
• コード長: 1,011 bytes
• コンパイル時間: 290 ms
• コンパイル使用メモリ: 30,848 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-02 09:09:29
• 合計ジャッジ時間: 1,920 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	6 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	7 ms; 5,376 KB
testcase_05	AC	6 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	5 ms; 5,376 KB
testcase_09	AC	5 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB

ソースコード

// yuki 942 プレゼント配り
// 2019.12.5 bal4u

#include <stdio.h>

typedef long long ll;

int getchar_unlocked(void);
int putchar_unlocked(int c);
#define gc() getchar_unlocked()
#define pc(c) putchar_unlocked(c)

int in() {   // 非負整数の入力
	int n = 0, c = gc();
	do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0');
	return n;
}

void out(int n) { // 非負整数の表示（出力）
	int i = 0; char b[30];
	while (n) b[i++] = n % 10 + '0', n /= 10;
	while (i--) pc(b[i]);
}

void outs(char *s) { while (*s) pc(*s++); pc('\n'); }

int main()
{
	int N, K, a, b, s, x, f;
	
	N = in(), K = in();
	if (K == 1) {
		outs("Yes");
		a = 1, x = N; while (--x) out(a++), pc(' ');
		out(N), pc('\n');
	} else {
		s = N/K;
		if ((s & 1) || (ll)N*(N+1)%(2*K)) outs("No");
		else {
			outs("Yes");
			a = 1, b = N, s = N/K;
			while (K--) {
				f = 1, x = s; while (--x) {
					if (f) out(a++), pc(' '), f = 0;
					else   out(b--), pc(' '), f = 1;
				}
				out(b--), pc('\n');
			}
		}
	}
	return 0;
}