EOJ 1814 完全二叉树？

#include <iostream>
#include <stdlib.h>
#include <stdio.h>
int main()
{
    int n,r;
    scanf("%d%d",&n,&r);//共有n个节点，r为树根
    int i;
    int degree[1030],a,b,flag=0;//flag用来判断最终树的状态
    memset(degree,0,sizeof(degree));//将保存节点的度的数组初始化为0
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        degree[a]++;//每次输入，若出现了某个节点，就使他的度数加一
        degree[b]++;
    }
    if(n!=1)//如果节点数目不为一
    {
        if(degree[r]!=2)
            flag=1;//如果根节点度数不为2，说明不是完全二叉树，flag置为1
        for(i=1;i<=n;i++)//对所有的节点遍历
        {
            if((i!=r)&°ree[i]!=3&°ree[i]!=1)//如果除了根节点其他的节点的度数不为1或者不为3，则不是完全二叉树
            {
                flag=1;
                break;
            }
        }
    }
    if(flag==1)//flag为1时，不是完全二叉树
        printf("no\n");
    else printf("yes\n");
}

一开始出错是因为判断非叶子节点时，只考虑了它的出度而没有去想入度，所以认为当他的度为二时为完全二叉树，后来加入了入度。注意先将有特殊情况的根节点排除，再判断其他的节点。