刚学OI,#9MLE,求大佬改题

P1197 [JSOI2008] 星球大战

FiresonZ @ 2022-05-21 21:51:08

#include<iostream>
#include<array>
#include<algorithm>
#include<cmath>
#include<bitset>
#include<vector>
#define reg register
using namespace std;
using gg=long long;
struct node{
    gg from,to,nxt;
};
array<gg,200001> x,y;
array<gg,200010> fa,mt;
array<gg,200010> res;
array<gg,200010> head;
array<node,400010> plt;
array<bool,200010> ned;
gg top=0;
gg fafind(const gg &x)
{
    if(fa[x]==x)
    {
        return x;
    }
    return fa[x]=fafind(fa[x]);
}
void add(const gg &from,const gg &to)
{
    top++;
    plt[top].from=from;
    plt[top].to=to;
    plt[top].nxt=head[from];
    head[from]=top;
    return;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    reg gg n,m,k,ans;
    cin>>n>>m;
    for(reg gg i=0;i<n;i++) fa[i]=i;
    for(reg gg i=1;i<=m;i++)
    {
        cin>>x[i]>>y[i];
    }
    cin>>k;
    ans=n-k;
    for(reg gg i=1;i<=k;i++)
    {
        cin>>mt[i];
        ned[mt[i]]=true;
    }
    for(reg gg i=1;i<=m;i++)
    {
        if(ned[x[i]]==false&&ned[y[i]]==false)
        {
            gg fidx=fafind(x[i]),fidy=fafind(y[i]);
            if(fidx!=fidy)
            {
                ans--;
                fa[fidx]=fidy;
            }
        }
        else
        {
            add(x[i],y[i]);
            add(y[i],x[i]);
        }
    }
    for(reg gg i=k;i>=0;i--)
    {
        res[i]=ans;
        ned[mt[i]]=false;
        ans++;
        gg fidmt=fafind(mt[i]);
        for(reg gg j=head[mt[i]];j;j=plt[j].nxt)
        {
            gg fidplt=fafind(plt[j].to);
            if(ned[plt[j].to]==false&&fidmt!=fidplt)
            {
                ans--;
                if(ans<1)
                {
                    ans=1;
                }
                fa[fidplt]=fidmt;
            }
        }
    }
    for(reg gg i=0;i<=k;i++) cout<<res[i]<<'\n';
    return 0;
}

by I_like_magic @ 2023-08-18 15:04:15

@FiresonZ 这道题其实不用 long long


|