こんばんは.初めて質問します.
この度は,何度もソースコードを見直したのですが,どこが間違っているかわからなかったため質問いたします
AtcoderRegularContest037 B問題「バウムテスト」についての質問なのですが,
以下のソースコードで提出してもWAが出てしまいます.
この問題はサンプル以外のテストパターンが公開されておらず,
原因を突き止めることができなかったので,問題があれば指摘していただきたいです.
該当のソースコード
C++
1#include<bits/stdc++.h> 2using ll= long long; 3#define REP(i,n) for(ll i=0;i<ll(n);i++) 4 5using namespace std; 6/*..................DEFINE GLOBAL VARIABLES...................*/ 7int N,M; 8 9int par[1000]; 10int ht[1000]; 11/*.....................DEFINE FUNCTIONS ......................*/ 12void init(int n){ 13 REP(i,n){ 14 par[i]=i; 15 } 16} 17 18int root(int x){ 19 if(par[x]==x) return x; 20 else return par[x]=root(par[x]); 21} 22 23void unite(int x,int y){ 24 x=root(x); 25 y=root(y); 26 if(x==y) return; 27 if(ht[x] < ht[y]){ 28 par[x] = y; 29 }else{ 30 par[y] = x; 31 if(ht[x] == ht[y]){ 32 ht[x]++; 33 } 34 } 35} 36 37bool same(int x,int y){ 38 return root(x)==root(y); 39} 40/*.........................kemkemG0...........................*/ 41int main() { 42 cin>>N>>M; 43 44 int ng[1000]; 45 46 init(110); 47 48 REP(i,M){ 49 int v,w; 50 cin>>v>>w; 51 v--;w--; 52 if(same(v,w)){ 53 ng[root(v)]=-1; 54 continue; 55 } 56 else unite(v,w); 57 } 58 59 REP(i,N){ 60 if(ng[i]==-1) ng[root(i)]=-1; 61 } 62 63 int count=0; 64 65 for(int i=0;i<N;i++){ 66 if(root(i)==i&&ng[i]==0) count++; 67 } 68 69 cout<<count<<endl; 70} 71 72
考えたこと
0.union-find木を初期化する.
1.点たちをつなぐ.既に同じ木にいる2点の間を結ぶとき閉路ができるはずなので,その根xに対してng[x]=-1としておく.(このときの根は最終的に根じゃないかもしれないので,次の操作2をしています)
2.ng[i]=-1となる点を持つ木の根のngも-1とする.
3.root(i)=iかつng[i]=0となる点が木の個数であるからそれを出力する.
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/11/30 12:17