Rikka with Tree （模拟过程（树的特殊性））

题目来源：https://vjudge.net/problem/HDU-5423
【题意】
众所周知，萌萌哒六花不擅长数学，所以勇太给了她一些数学问题做练习，其中有一道是这样的：
对于一棵树TT,令F(T,i)F(T,i)为点1到点ii的最短距离（边长是1）.
两棵树AA和BB是相似的当且仅当他们顶点数相同且对于任意的ii都有F(A,i)=F(B,i)F(A,i)=F(B,i).
两棵树AA和BB是不同的当且仅当他们定点数不同或者存在一个ii使得以1号点为根的时候ii在两棵树中的父亲不同。
一棵树AA是特殊的当且仅当不存在一棵和它不同的树和它相似。
现在勇太想知道一棵树到底是不是特殊的。
对于每一组数据，如果给定树是特殊的输出”YES”否则输出”NO”。
【思路】
首先要搞清树与树之间什么样的是相似，什么是不同，两者结合，变成了特殊。而这个特殊的树经过推理之后就知道，若是特殊的，就必须是111111….x型，也就是说，一个父亲节点一旦儿子数量超过一个，那么他的儿子就不能再有儿子。根据这个特点，我选择了用数组，一维数组记录儿子节点是父亲节点的第几个儿子，而b数组记录的是i的第j个儿子是某某某。模拟整个过程。
【代码】

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long LL;
int a[1010];
int b[1010][1010];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        int x,y;
        for(int i=1;iscanf("%d%d",&x,&y);
            a[x]++;
            b[x][a[x]]=y;
        }
        int flag=0;
        for(int i=1;i<=n;i++)
        {
            if(a[i]>1)
            {
                for(int j=1;j<=a[i];j++)
                {
                    if(a[b[i][j]]>=1)
                    {
                        flag=1;
                        break;
                    }
                }
                if(flag)
                    break;
            }
        }
        if(flag)
            printf("NO\n");
        else
            printf("YES\n");
    }
}