[AtCoder]ARC037 B問題バウムテスト

Question

前提・実現したいこと

ARC037 B問題にて、私のコードでWAになるようなテストケースが知りたいです。
サンプルケースは全て通るのですが、他のケースで通らないようです。よろしくお願いいたします。

該当のソースコード

#include<iostream>
#include<vector>

using ll = long long;
int n, m;
using graph = std::vector<std::vector<int>>;
graph g;
std::vector<int> color;

bool dfs(int u, int c, int pre = -1){
    bool flag = false;
    color[u] = c;
    for(auto v: g[u]){
        if(color[v] == -1){
            flag = flag || dfs(v, c, u);
        }else if(pre != v && color[v] == c){
            // 閉路あり
            flag = true;
        }
    }

    return flag;
}

int main(){
    std::cin >> n >> m;
    g.resize(n);
    color.assign(n, -1);
    for(int i = 0; i < m; i++){
        int u, v;
        std::cin >> u >> v;
        u--; v--;
        g[u].push_back(v);
        g[v].push_back(u);
    }

    int ans = 0;
    int c = 0;
    for(int i = 0; i < n; i++){
        if(color[i] == -1){
            bool flag = dfs(i, c);
            if(!flag) ans++;
            c++;
        }
    }

    std::cout << ans << std::endl;
    return 0;
}

試したこと

閉路検出の部分で問題があるだろうと思い、他の方のコードを参考に見てみました。そこでコード中の「閉路あり」というコメントがついている部分の判定をpre != v && color[v] != -1に変更したところACとなりました。今いる頂点の色c以外の別の色が既についるところに向かう辺は無い（なぜならそのような行先があればその行先の色を扱っているときに今いる頂点に来ているはずだから）と考えているのですが、どうやらこれが違っているのでしょうか。

Answer 1

4 4
1 2
2 3
3 1
2 4

小さいケースなので動きを追えば何が起きてるかわかると思います。

Comments

[退会済みユーザー]

`flag`が一度trueになってしまうと`flag || dfs`と書いたせいで色付けしない部分が出てしまうことが分かりました。回答ありがとうございました。