PTA 1065 单身狗 (c语言)

1065 单身狗 (25 分)

分析：我们规定无婚配为空白（-1）如果已婚/情侣中的一个人到场称为签到（-2）如果是单身狗则为（-3）
首先我们知道id是由0000到9999，所以我们建立一个长度为100000的数组。在输入婚配/情侣id的时候让他们互相映射。
然后再输入在场的人数的id的时候，在进行处理。代码如下：

#define BLANK -1
#define SIGNED -2
#define SINGLE -3
int main()
{
    int couple[100000],count=0,N,M,ID1,ID2;
    for(int i=0;i<100000;i++)
        couple[i]=BLANK;
    scanf("%d",&N);
    for(int i=0;i<N;i++)
    {
        scanf("%d %d",&ID1,&ID2);
        couple[ID1]=ID2;
        couple[ID2]=ID1;
    }
    scanf("%d", &M);
    for(int i=0;i<M;i++)          
    {
        scanf("%d",&ID1);
        if(couple[ID1]>=0)               
            couple[ID1]=SIGNED;           
        else                                
            couple[ID1]=SINGLE,count++;  
    }
    for(int i=0;i<100000;i++)
        if(couple[i]>=0&&couple[couple[i]]==SIGNED)
            couple[couple[i]]=SINGLE,count++;
    printf("%d\n",count);
    for(int i=0;i<100000;i++)
        if(couple[i]==SINGLE)
            printf("%05d%c",i,--count?' ':'\0');
    return 0;
}