codeforces 791B Bear and Friendship Condition

题目链接：http://codeforces.com/contest/791/problem/B

题意：n个人，共有m对关系。如果x和y有关系，y和z有关系，只有x和y有关系的时候才认为这样是合法的。现在问你输入是否合法。

分析：我是dfs做的，将每对关系存入vector，然后数组记录每一个人他一共有多少关系，然后dfs他，看和他有关系的人是不是关系数也和他一样。一旦有不一样的，就输出"NO".

AC代码：

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

int a[150005];
vector<int>g[150005];
int vis[150005];
int p=0,number;
void dfs(int x){
    int d=g[x].size();
    if(p==1) return;
    vis[x]=1;
    number++;
    for(int i=0;i<d;i++){
        if(vis[g[x][i]]==0) {
            if(a[g[x][i]]!=a[x]){
                p=1;
                break;
            }
            dfs(g[x][i]);
        }
    }
}
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n,m;
    memset(a,0,sizeof(a));
    cin>>n>>m;
    int x,y;
    for(int i=1;i<=m;i++){
        cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
        a[x]++;
        a[y]++;
    }
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++){
        number=0;
        if(vis[i]==0) {
            dfs(i);
            if(p==1||number!=a[i]+1) {p=1;break;}
        }
    }
    if(p==0) cout<<"YES"<<endl;
    else cout<<"NO"<<endl;

return 0;
}