結果

問題 No.2998 Rainbow Christmas Tree
コンテスト
ユーザー hiromi_ayase
提出日時 2025-11-13 18:57:52
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,282 bytes
コンパイル時間 6,522 ms
コンパイル使用メモリ 335,536 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-11-13 18:58:21
合計ジャッジ時間 27,812 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35 WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

#include <atcoder/all>
using namespace std;
using i32 = int;
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
#define FAST_IO                \
  ios::sync_with_stdio(false); \
  cin.tie(0);
const i64 INF = 1001001001001001001;
using Modint = atcoder::static_modint<998244353>;

void solve1(vector<vector<int>>& P, int root) {}

int main() {
  FAST_IO

  int N, K;
  cin >> N >> K;
  vector P(N - 1, vector(N, 0));
  for (int i = 0; i < N - 1; i++) {
    for (int j = 0; j < N; j++) {
      cin >> P[i][j];
      P[i][j] --;
    }
  }

  set<int> S1, S2;
  S1.insert(0);
  S2.insert(N - 1);

  set<int> g1, g2;
  for (int i = 0; i < K; i ++) {
    g1.insert(i);
  }
  for (int i = K; i < N - 1; i ++) {
    g2.insert(i);
  }

  vector<int> Q(N, -1);

  while (K != 0 && K != N - 1) {
    bool flg = false;

    if (g1.size() > 0) {
      auto g = *g1.begin();
      g1.erase(g);

      for (int to = 0; to < N; to ++) {
        int from = P[g][to];
        if (from < 0) continue;

        if (S2.contains(to) && !S2.contains(from)) {
          if (S1.contains(from)) flg = true;
          S2.insert(from);
          Q[to] = g;
          break;
        }
      }
    } else {
      auto g = *g2.begin();
      g2.erase(g);

      for (int to = 0; to < N; to ++) {
        int from = P[g][to];
        if (from < 0) continue;

        if (S1.contains(to) && !S1.contains(from)) {
          if (S2.contains(from)) flg = true;
          S1.insert(from);
          Q[to] = g;
          break;
        }
      }
    }
    if (flg) break;
  }

  set<int> S;
  if (K != 0) {
    for (auto s : S1) S.insert(s);
  }
  if (K != N - 1) {
    for (auto s : S2) S.insert(s);
  }

  set<int> G;
  for (auto g : g1) G.insert(g);
  for (auto g : g2) G.insert(g);

  while (G.size() > 0) {

    int found = -1;
    for (int g : G) {
      for (int to = 0; to < N; to ++) {
        int from = P[g][to];
        if (from < 0) continue;
        if (S.contains(from) && !S.contains(to)) {
          S.insert(to);
          Q[to] = g;
          found = g;
          break;
        }
      }
      if (found >= 0) break;
    }
    G.erase(found);
  }

  cout << "Yes" << endl;
  for (auto e : Q) {
    cout << (e + 1) << " ";
  }
  cout << endl;
}
0