请问这份代码有什么毛病

P1197 [JSOI2008] 星球大战

wweiyi @ 2021-06-17 17:19:54

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;
const int maxn=200005;
//建图以及并查集 
struct edge{
    int e,next;
}ed[maxn*2];
int first[maxn],en;
void add_edge(int s,int e)
{
    en++;
    ed[en].next=first[s];
    first[s]=en;
    ed[en].e=e;
}
int f[100010];
void init()
{
    for(int i=1;i<=100000;i++)
    f[i]=i;
}
int get(int x)
{
    return f[x]=(x==f[x]? x:get(f[x]));
}
void merge(int x,int y)
{
    f[get(x)]=get(y);
}
int n,m,k;
int tot=0,ans[maxn];
int hit[maxn];
bool ishit[maxn]={false},isused[maxn]={false};
void is_c()
{
    tot++;
    for(int i=0;i<n;i++)
    {
        if(!isused[get(f[i])])
        {
            isused[get(f[i])]=true;
            ans[tot]++;
        }
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    init();
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        cin>>x>>y;
        add_edge(x,y),add_edge(y,x);
    }
    cin>>k;
    for(int i=1;i<=k;i++)
    {
        cin>>hit[i];
        ishit[hit[i]]=true;
    }
    reverse(hit+1,hit+1+k);
    /*
    for(int i=1;i<=k;i++)
    cout<<hit[i]<<endl;
    */
    for(int i=0;i<n;i++)
    {
        if(!ishit[i])
        for(int j=first[i];j;j=ed[j].next)
        {
            int e=ed[j].e;
            if(!ishit[e])
            merge(i,e);
        }
    }
    for(int i=1;i<=k;i++)
    {
        memset(isused,false,sizeof(isused));
        for(int j=i;j<=k;j++)
        isused[hit[j]]=true;
        is_c();
        for(int j=first[hit[i]];j;j=ed[j].next)
        {
            int e=ed[j].e;
            merge(hit[i],e);
        }
    }
    is_c();
    for(int i=tot;i>0;i--)
    cout<<ans[i];
    //cout<<ans[tot+1];
    return 0;
}

|