PTA 密接判定v1（close_contact1）

Omicron变种病毒传染性太强了，城市屡屡破防。封闭在家的我决定为防疫做点事：帮助筛选密切接触、次密切接触人员。
网格员提供了一些小区的风险时段内的接触记录、确诊人员，需要判断人员的风险分类：

• A类：疑似病例、临床诊断病例、确诊病例。（输入数据）
• B类：与A类人员有接触的人员。（需要判断）
• C类：与B类人员有接触的人员。（需要判断）
• D类：其他人员。

输入规格

• 输入的数据有多组，处理至EOF为止。
• 每组数据
  • 首行有4项：日期D（字符串）、小区人数N、接触记录数量C、确诊人数P，间隔1个空格。N、C、P都是非负整数。
  • 之后是C行接触记录。每条记录由两个人员ID组成，间隔1个空格。没有重复的接触记录。
  • 然后是P行确诊人员ID。
  • 为保护隐私，发给我的数据中人员ID已经脱敏，都是些序号，范围从1到N。
• 小区最多2000人。本题不考察时空复杂度。

输出规格

• 对每组数据：
  • 输出日期，换行。
  • 按照序号由小到大逐行输出A、B、C类人员ID和类别（A/B/C字母），间隔1个空格。D类略过。

样例输入

2022-04-01 8 5 2
2 1
8 5
5 2
3 4
4 2
1
8
2022-01-01 5 0 0

样例输出

2022-04-01
1 A
2 B
4 C
5 B
8 A
2022-01-01

样例解释

• 第1组：1、8为A类。2、5与A类人员有接触，属于密接B类。4与B类人员有接触，属于次密接C类。
• 第2组：为空。

提示

• 最多2K人，接触记录排除重复后最多2K * 2K ~ 4M条，可以用布尔型的二维数组bool contact[N][N];记录，当contact[u][v]为true时表示编号u和v的人员有接触。约需要4M字节，空间可行。
  • 是否接触其实用1-bit就能表示，1-byte能表示8条接触记录，所以只需4M / 8 ~ 512KB
• 人员类别可用数组char tag[N]保存，全部初始化为'D'或NUL。
  • 用'D'更清晰，可用memset(s, c, n)设置。
  • 用NUL更方便，写成char tag[N]{};即可。
  • 这两种方式如果不太理解，建议都试一遍。
• 样例第一组数据读入后的contact数组状态：
  • 为使区分明显，用.代表false，用T代表true。

  • 读入全部接触记录后，先根据A类把B类全标记出来，再根据B类从剩余人员中把C类标记出来。

 代码实现（方法1）：

#include<stdio.h>
#include<stdlib.h>
void test(int);
char date[15];
typedef struct ArcNode* Arc;
struct ArcNode
{
    int adjvex;
    Arc next;
};

struct Vnode
{
    Arc bgn;
    char type;
};
struct Vnode graph[2001];
void init(int N);
void createNode(int v1,int v2);
void sift(int N);
void Print(int N)
{
    for(int i=1;i<=N;i++)
    {
        if(graph[i].type!='D')
        {
            printf("%d %c\n",i,graph[i].type);
        }
    }
    return;
}
int main()
{
    int N,C,P;
    while(scanf("%s %d%d%d",date,&N,&C,&P)!=EOF)
    {
        init(N);
        for(int i=0;i<C;i++)
        {
            int v1, v2;
            scanf("%d%d",&v1,&v2);
            createNode(v1,v2);
            createNode(v2,v1);
        }
        for(int i=0;i<P;i++)
        {
            int v;
            scanf("%d",&v);
            graph[v].type='A';
        }
        sift(N);
        printf("%s\n",date);
        //test(1);
        Print(N);
    }
    return 0;
}
void init(int N)
{
    for(int i=1;i<=N;i++)
    {
        graph[i].bgn=NULL;
        graph[i].type='D';
    }
    return;
}
void createNode(int v1,int v2)
{
    Arc newnode=(Arc)malloc(sizeof(struct ArcNode));
    newnode->adjvex=v2;
    newnode->next=NULL;

    if(graph[v1].bgn==NULL)
    {
        graph[v1].bgn=newnode;
    }
    else
    {
        Arc pre=graph[v1].bgn;
        Arc cur=pre->next;
        while(cur!=NULL)
        {
            pre=cur;
            cur=cur->next;
        }
        pre->next=newnode;
    }
    return;
}
void sift(int N)
{
    for(int i=1;i<=N;i++)
    {//寻找B类型
        if(graph[i].type=='A')
        {
            Arc cur=graph[i].bgn;
            while(cur!=NULL)
            {
                int v=cur->adjvex;
                if(graph[v].type=='D')graph[v].type='B';
                cur=cur->next;
            }
        }
    }
    //Print(N);
    for(int i=1;i<=N;i++)
    {
        if(graph[i].type=='B')
        {
            Arc cur=graph[i].bgn;
            while(cur!=NULL)
            {
                int v=cur->adjvex;
                if(graph[v].type=='D')graph[v].type='C';
                cur=cur->next;
            }
        }
    }
    return;
}
void test(int i)
{//调试函数
    printf("Start to test:\n");
    Arc cur=graph[i].bgn;
    while(cur!=NULL)
    {
        printf("%d %d\n",i,cur->adjvex);
        cur=cur->next;
    }
    return;
}