結果

問題 No.679 不思議マーケット
コンテスト
ユーザー beet
提出日時 2018-04-27 22:36:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 2,088 bytes
コンパイル時間 2,081 ms
コンパイル使用メモリ 183,512 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-27 21:57:55
合計ジャッジ時間 2,844 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;

struct SCC{
  Int n;
  vector<vector<Int> > G,rG,T,rT,C;
  vector<Int> vs,used,belong;
  SCC(){}
  SCC(Int sz):n(sz),G(sz),rG(sz),used(sz),belong(sz){}
  
  void add_edge(Int from,Int to){
    G[from].push_back(to);
    rG[to].push_back(from);
  }
  
  void input(Int m,Int offset=0){
    Int a,b;
    for(Int i=0;i<m;i++){
      cin>>a>>b;
      add_edge(a+offset,b+offset);
    }
  }
  
  void dfs(Int v){
    used[v]=1;
    for(Int i=0;i<(Int)G[v].size();i++){
      if(!used[G[v][i]]) dfs(G[v][i]);
    }
    vs.push_back(v);
  }
  
  void rdfs(Int v,Int k){
    used[v]=1;
    belong[v]=k;
    C[k].push_back(v);
    for(Int i=0;i<(Int)rG[v].size();i++){
      if(!used[rG[v][i]]) rdfs(rG[v][i],k);
    }
  }
  
  Int build(){
    fill(used.begin(),used.end(),0);
    vs.clear();
    for(Int v=0;v<n;v++){
      if(!used[v]) dfs(v);
    }
    fill(used.begin(),used.end(),0);
    Int k=0;
    for(Int i=vs.size()-1;i>=0;i--){
      if(!used[vs[i]]){
	T.push_back(vector<Int>());
	rT.push_back(vector<Int>());
	C.push_back(vector<Int>());
	rdfs(vs[i],k++);
      }
    }
    for(Int i=0;i<n;i++)
      for(Int u:G[i])
	if(belong[i]!=belong[u]){
	  T[belong[i]].push_back(belong[u]);
	  rT[belong[u]].push_back(belong[i]);
	}
    for(Int i=0;i<k;i++){
      sort(T[i].begin(),T[i].end());
      T[i].erase(unique(T[i].begin(),T[i].end()),T[i].end());
      sort(rT[i].begin(),rT[i].end());
      rT[i].erase(unique(rT[i].begin(),rT[i].end()),rT[i].end());
    }
    return k;
  }
};

//INSERT ABOVE HERE
signed main(){
  Int n,m;
  cin>>n>>m;
  SCC scc(n);
  for(Int i=0;i<m;i++){
    Int v,k;
    cin>>v>>k;
    v--;
    for(Int j=0;j<k;j++){
      Int u;
      cin>>u;
      u--;
      scc.add_edge(v,u);
    }
  }
  Int k=scc.build();
  //cout<<k<<endl;
  vector<Int> ok(k,1);
  auto T=scc.rT;
  auto C=scc.C;
  for(Int i=k-1;i>=0;i--){
    if(C[i].size()!=1u) ok[i]=0;
    for(Int j:T[i])
      if(!ok[i]) ok[j]=0;
  }
  Int ans=0;
  for(Int i=0;i<k;i++) ans+=ok[i]*C[i].size();
  cout<<ans<<endl;
  return 0;
}
0