結果

問題 No.679 不思議マーケット
ユーザー beet
提出日時 2018-04-27 22:36:23
言語 C++14
(gcc 8.3.0)
結果
AC  
実行時間 10 ms
コード長 2,088 Byte
コンパイル時間 1,707 ms
使用メモリ 2,264 KB
最終ジャッジ日時 2019-09-09 05:28:58

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
input00.txt AC 3 ms
1,524 KB
input01.txt AC 3 ms
1,516 KB
input02.txt AC 3 ms
1,524 KB
input03.txt AC 10 ms
2,264 KB
input04.txt AC 6 ms
1,924 KB
input05.txt AC 8 ms
1,928 KB
input06.txt AC 5 ms
1,740 KB
input07.txt AC 4 ms
1,704 KB
input08.txt AC 6 ms
1,728 KB
input09.txt AC 3 ms
1,516 KB
input10.txt AC 5 ms
1,652 KB
input11.txt AC 4 ms
1,704 KB
input12.txt AC 5 ms
1,656 KB
input13.txt AC 5 ms
1,788 KB
input14.txt AC 5 ms
1,756 KB
input15.txt AC 4 ms
1,608 KB
input16.txt AC 3 ms
1,520 KB
input17.txt AC 3 ms
1,516 KB
テストケース一括ダウンロード

ソースコード

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