結果

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

ソースコード

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<vector<int>> ng(cnt);
  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];
      ng[a].push_back(b);
    }
  }
  check.assign(n, 0);
  topo.clear();
  g = ng;
  for (int i = 0; i < cnt; ++i)
  {
    topologicalsort(topologicalsort, i);
  }
  reverse(topo.begin(), topo.end());
  if (topo[0] != newid[0])
  {
    cout << "No" << endl;
    return 0;
  }
  for (int i = 0; i < topo.size() - 1; ++i)
  {
    int np = topo[i + 1];
    bool flag = true;
    for (int v : ng[topo[i]])
    {
      if (v == np)
      {
        flag = false;
        break;
      }
    }
    if (flag)
    {
      cout << "No" << endl;
      return 0;
    }
  }
  cout << "Yes" << endl;
}
0