結果

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

ソースコード

diff # 

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

using namespace std;

int main()
{
  int n, m;
  cin >> n;
  vector<vector<int>> g(n), gr(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);
      gr[v].push_back(i);
    }
  }
  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);
  reverse(topo.begin(), topo.end());
  vector<int> tmp;
  auto dfs = [&](auto f, int now) -> void
  {
    if (check[now] == 0) return;
    tmp.push_back(now);
    check[now] = 0;
    for (int v : gr[now]) f(f, v);
  };
  vector<int> newid(n);
  int cnt = 0;
  for (int v : topo)
  {
    if (check[v] == 0) continue;
    tmp.clear();
    dfs(dfs, v);
    for (auto v1 : tmp) 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