刚学OI 求大佬改题 WA#9

P1197 [JSOI2008] 星球大战

mistakes @ 2021-03-20 16:54:34

#include<iostream>
#include<vector>
using namespace std;
#define N 500500
vector<int> map[N];
bool ma[N];
int fa[N],q[N];
int n,m;
int ans=0;
int getfa(int x)
{
    if(fa[x]==x) return x;
    else return fa[x]=getfa(fa[x]);
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        cin>>x>>y;
        map[x].push_back(y);
        map[y].push_back(x);
    }
    int k;
    cin>>k;
    int ans[N];
    int tot=n-k;
    for(int i=0;i<k;i++)
        cin>>q[i],ma[q[i]]=1;
    for(int i=0;i<m;i++)
    {
        int u=i;
        if(ma[u]) continue;
        for(int j=0;j<map[u].size();j++)
        {
            if(ma[map[u][j]]) continue;
            if(getfa(u)!=getfa(map[u][j])) fa[getfa(u)]=fa[getfa(map[u][j])],tot--;
        }
    }
    ans[k]=tot;
    for(int i=k-1;i>=0;i--)
    {
        int u=q[i];
        tot++;
        ma[u]=0;
        for(int j=0;j<map[u].size();j++)
        {
            if(ma[map[u][j]]) continue;
            if(getfa(map[u][j])!=getfa(u)) fa[getfa(map[u][j])]=fa[getfa(u)],tot--;
        }
        ans[i]=tot;
    }
    for(int i=0;i<=k;i++)
        cout<<ans[i]<<endl;
    return 0;
}

by HerikoDeltana @ 2021-03-21 10:28:05

震惊,居然是MTK,我说这个标题那么熟悉......


|