白駿13023.ABCDE

8615 ワード
アルゴリズム#アルゴリズム# 白駿 テキストリンク

13023. ABCDE


N人と友達の関係数Mをあげます.この場合、グラフで友達関係を表す場合は、グラフの深さが5以上かどうかを判断します.
人の数と友人関係の数は2000に制限されているため,すべてのノードに対してDFSメソッドチェックを行ってもタイムアウトは発生しない.
#include <iostream>
#include <vector>
#define vi vector<int>
using namespace std;

vector<vi> edge;
vector<bool> is_visit;

bool dfs(int cur, int depth){
    if(depth == 4) return true;
    is_visit[cur] = true;
    bool ret = false;
    for(int i=0; i<edge[cur].size() and !ret; i++){
        int next = edge[cur][i];
        if(is_visit[next]) continue;
        ret = dfs(next, depth + 1);
    }
    is_visit[cur] = false;
    return ret;
}

int main(){
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    int n, m; cin>>n>>m;

    edge = vector<vi>(n);
    is_visit = vector<bool>(n);
    while(m--){
        int s, e; cin>>s>>e;
        edge[s].push_back(e);
        edge[e].push_back(s);
    }

    bool is_friend = false;
    for(int i=0; i<n and !is_friend; i++){
        fill(is_visit.begin(), is_visit.end(), false);
        is_friend = dfs(i, 0);
    }
    cout<<is_friend;
}
タイプスクリプト...rest, destructuring
白駿-WElove kriiiii[10718]