結果
問題 | No.942 プレゼント配り |
ユーザー |
![]() |
提出日時 | 2019-12-05 01:28:37 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 281 ms / 2,000 ms |
コード長 | 1,439 bytes |
コンパイル時間 | 2,587 ms |
コンパイル使用メモリ | 80,324 KB |
実行使用メモリ | 52,680 KB |
最終ジャッジ日時 | 2024-12-14 22:50:56 |
合計ジャッジ時間 | 8,458 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
import java.io.PrintWriter; import java.util.ArrayList; import java.util.List; import java.util.Scanner; public class Main { public static void main(String[] args) throws Exception { Scanner sc = new Scanner(System.in); int n = sc.nextInt(); int k = sc.nextInt(); sc.close(); long n2 = (long) n * (n + 1) / 2; if (n > 1 && n == k || n2 % k != 0) { System.out.println("No"); return; } PrintWriter pw = new PrintWriter(System.out); pw.println("Yes"); List<List<Integer>> list = new ArrayList<>(k); for (int i = 0; i < k; i++) { list.add(new ArrayList<>()); } if (n / k % 2 == 1) { int a = k; for (int i = 0; i < k; i++) { list.get(i).add(a); a -= 2; if (a < 1) { a = k - 1; } } a = k + k / 2; for (int i = 0; i < k; i++) { a++; list.get(i).add(a); if (a == 2 * k) { a = k; } } for (int i = 2 * k; i < n; i++) { if (i / k % 2 == 0) { list.get(i % k).add(i + 1); } else { list.get(k - i % k - 1).add(i + 1); } } } else { for (int i = 0; i < n; i++) { if (i / k % 2 == 0) { list.get(i % k).add(i + 1); } else { list.get(k - i % k - 1).add(i + 1); } } } for (int i = 0; i < k; i++) { StringBuilder sb = new StringBuilder(); for (int j : list.get(i)) { sb.append(j).append(' '); } sb.deleteCharAt(sb.length() - 1); pw.println(sb.toString()); } pw.flush(); } }