PTA 天梯赛7-3 密接判定v1

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人，接触记录最多4M条，可以用布尔型的二维数组bool contact[N][N];记录，当contact[u][v]为true时表示编号u和v的人员有接触。

错在第三个测试点，目前看不出哪里需要更改。

c++刚刚起步，很多都不会，麻烦大佬们指点指点。

#include
#include
using namespace std;
int main(void){
    char s[100];
    bool contact[2000][2000];
    char b[10005];
    int n,c,p;
    while(scanf("%s",&s)!=EOF){
        scanf("%d %d %d",&n,&c,&p);
        getchar();
        if(c==0){
            printf("%s\n",s);
            continue;
        }
        int y=0,x=0;
        for(int i=0;i<c;i++){
            scanf("%d %d",&y,&x);
            getchar();
            contact[y][x]=1;
        }
        int a[10005];
        int c[10005];
        int l=0;
        for(int i=1;i<=p;i++){
            scanf("%d",&a[i]);
            getchar();
            b[a[i]]='A';
            for(int j=1;j<=n;j++){
                if((contact[a[i]][j]==1&&b[j]!='A')||(contact[j][a[i]]==1&&b[j]!='A')){
                    b[j]='B';
                    c[l]=j;
                    l++;
                }
            }
        }
        for(int i=0;i<l;i++){
            for(int j=1;j<=n;j++){
                if((contact[c[i]][j]==1&&b[j]!='A'&&b[j]!='B')||(contact[j][c[i]]==1&&b[j]!='A'&&b[j]!='B')){
                    b[j]='C';
                }
            }
        }
        printf("%s\n",s);
        for(int i=1;i<=n;i++){
            if(b[i]=='A'||b[i]=='B'||b[i]=='C'){
                printf("%d %c\n",i,b[i]);
            }
        }
        for(int i=0;i<=n;i++){
            a[i]=0;
            c[i]=0;
            b[i]='0';
        }
        for(int i=0;i<=1000;i++){
            for(int j=0;j<1000;j++){
                contact[i][j]=0;
                contact[j][i]=0;
            }
        }
    }
    return 0;
}