PAT乙级 1065 单身狗 (25 分) 思维 模拟 测试点3 4

代码传送门
3 4 超时就是单纯因为数据太大了，如果做法超时那只能换个做法了
这道题我参考了柳神的代码

for(i=0;i<n;i++){
            scanf("%d %d",&c,&d);
            a[c]=d;
            a[d]=c;
        }

这一段很巧妙，大一学过，但做题真的没想到，一开始写的太复杂了string+结构体+两重for 时间复杂度爆炸
超时代码：没过测试点3 4

#include <iostream>
#include<stdio.h>
#include<algorithm>
#include<string>
#include<string.h>
using namespace std;
string a[10010];
int flag1[10010];
struct node{
    string n1;
    string n2;
    int flag=0;
    int prei;
}b[50010];

int main()
{
    int n,m,i,j;
    while(~scanf("%d",&n)){
            memset(flag1,0,sizeof(flag1));
        for(i=0;i<n;i++){
            cin>>b[i].n1>>b[i].n2;
        }
        scanf("%d",&m);
        for(i=0;i<m;i++){
            cin>>a[i];
        }
        int ans=m;
        for(i=0;i<m;i++){
            for(j=0;j<n;j++){
                if(a[i]==b[j].n1&&b[j].flag==2){
                    flag1[i]=1;
                    flag1[b[j].prei]=1;
                    //ans=ans-1;
                }
                else if(a[i]==b[j].n2&&b[j].flag==1){
                    flag1[i]=1;
                    flag1[b[j].prei]=1;
                    //ans=ans-1;
                }
                else if(a[i]==b[j].n1&&b[j].flag==0){
                    b[j].flag=1;
                    b[j].prei=i;
                }
                else if(a[i]==b[j].n2&&b[j].flag==0){
                    b[j].flag=2;
                    b[j].prei=i;
                }

            }
        }
        int anf=0;
        string an[10010];
        //printf("%d\n",ans);
        for(i=0;i<m;i++){
           if(flag1[i]!=1){
            an[anf]=a[i];
            anf++;
            //cout<<a[i]<<endl;
           }

        }

        sort(an,an+anf);
        //cout<<an[0];
        cout<<anf<<endl;
        for(i=0;i<anf;i++){
            if(i!=0)
                printf(" ");
            cout<<an[i];
        }


    }

    return 0;
}

后来重新写了一个 上图代码全删了

#include <iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<string>
#include<set>
#include<iterator>
using namespace std;
int a[500010],b[10010];
set<int>s;
int main()
{
    int n,m,i,j,c,d;
    while(~scanf("%d",&n)){
        memset(a,-1,sizeof(a));
        for(i=0;i<n;i++){
            scanf("%d %d",&c,&d);
            a[c]=d;
            a[d]=c;
        }
        scanf("%d",&m);
        for(i=0;i<m;i++)
            scanf("%d",&b[i]);
        for(i=0;i<m;i++){
            int cnt=0;
            for(j=0;j<m;j++){
                if(a[b[i]]!=b[j])cnt++;
                if(cnt==m)
                    s.insert(b[i]);
            }
        }
        printf("%d\n",s.size());
        for(auto it=s.begin();it!=s.end();it++){
            if(it!=s.begin())
                printf(" ");
            printf("%05d",*it);
        }
    }
    return 0;
}