結果

問題 No.2780 The Bottle Imp
ユーザー テナガザル
提出日時 2024-06-07 23:16:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,657 bytes
コンパイル時間 1,249 ms
コンパイル使用メモリ 91,228 KB
最終ジャッジ日時 2025-02-21 20:35:15
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 38 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>

using namespace std;

int main()
{
  int n, m;
  cin >> n;
  vector<vector<int>> g(n);
  for (int i = 0; i < n; ++i)
  {
    int m;
    cin >> m;
    for (int j = 0; j < m; ++j)
    {
      int v;
      cin >> v;
      --v;
      g[i].push_back(v);
    }
  }
  vector<int> check(n), topo;
  auto topologicalsort = [&](auto f, int now) -> void
  {
    if (check[now]) return;
    check[now] = 1;
    for (auto v : g[now]) f(f, v);
    topo.push_back(now);
  };
  for (int i = 0; i < n; ++i)
  {
    topologicalsort(topologicalsort, i);
  }
  vector<int> gr;
  auto dfs = [&](auto f, int now) -> void
  {
    if (check[now] == 0) return;
    gr.push_back(now);
    check[now] = 0;
    for (int v : g[now]) f(f, v);
  };
  vector<int> newid(n);
  int cnt = 0;
  for (int v : topo)
  {
    if (check[v] == 0) continue;
    gr.clear();
    dfs(dfs, v);
    for (auto v1 : gr) newid[v1] = cnt;
    ++cnt;
  }
  set<pair<int, int>> st;
  vector<int> np(cnt, -1);
  for (int i = 0; i < n; ++i)
  {
    for (auto v : g[i])
    {
      if (newid[i] == newid[v] || st.find({newid[i], newid[v]}) != st.end()) continue;
      st.insert({newid[i], newid[v]});
      int a = newid[i], b = newid[v];
      if (np[a] != -1)
      {
        cout << "No" << endl;
        return 0;
      }
      np[a] = b;
    }
  }
  vector<int> ncheck(cnt, 1);
  int now = newid[0];
  ncheck[now] = 0;
  while (np[now] != -1)
  {
    now = np[now];
    ncheck[now] = 0;
  }
  for (int i = 0; i < cnt; ++i)
  {
    if (ncheck[i])
    {
      cout << "No" << endl;
      return 0;
    }
  }
  cout << "Yes" << endl;
}
0