每日一题 牛客基础 第六场 动态最小生成树 最小生成树 线段树

#include<bits/stdc++.h>
using namespace std;
int n,m,q;
typedef vector<int> vec;
int dsu[205];
int fnd(int x)
{
    if(dsu[x]==x)return x;
    return dsu[x]=fnd(dsu[x]);
}
int merg(int x,int y)
{
    x=fnd(x),y=fnd(y);
    dsu[x]=y;
    return x!=y;
}
struct EDGE
{
    int a,b,w;
}e[30005];
struct node
{
    int l,r,v;
    vec ve;
} sgt[120005];
void pushup(int &v,vec &u,vec &a,vec &b)
{
    v=0;
    for(int i=1;i<205;i++)dsu[i]=i;
    int i=0,j=0;
    u.clear();
    while(i<a.size()||j<b.size())
    {
        if(i<a.size()&&j<b.size())
        {
            if(e[a[i]].w<e[b[j]].w)
            {
                if(merg(e[a[i]].a,e[a[i]].b))
                    u.push_back(a[i]),v+=e[a[i]].w;
                i++;
            }
            else
            {
                if(merg(e[b[j]].a,e[b[j]].b))
                    u.push_back(b[j]),v+=e[b[j]].w;
                j++;
            }
        }
        else if(i<a.size())
        {
            if(merg(e[a[i]].a,e[a[i]].b))
                    u.push_back(a[i]),v+=e[a[i]].w;
            i++;
        }
        else if(j<b.size())
        {
            if(merg(e[b[j]].a,e[b[j]].b))
                    u.push_back(b[j]),v+=e[b[j]].w;
            j++;
        }
    }
}
void iniSgt(int p,int l,int r)
{
    sgt[p].l=l;
    sgt[p].r=r;
    if(l==r)
    {
        sgt[p].ve.push_back(r);
        sgt[p].v=e[r].w;
        return;
    }
    int mid=(l+r)/2;
    iniSgt(p*2,l,mid);
    iniSgt(p*2+1,mid+1,r);
    pushup(sgt[p].v,sgt[p].ve,sgt[p*2].ve,sgt[p*2+1].ve);
}
void modify(int p,int X)
{
    int l=sgt[p].l,r=sgt[p].r;
    if(l==r&&l==X)
    {
        sgt[p].v=e[X].w;
        return;
    }
    int mid=(l+r)/2;
    if(X<=mid)
        modify(p*2,X);
    if(X>=mid+1)
        modify(p*2+1,X);
    pushup(sgt[p].v,sgt[p].ve,sgt[p*2].ve,sgt[p*2+1].ve);
}
int query(int p,int x,int y,vec &u)
{
    int l=sgt[p].l,r=sgt[p].r;
    if(x<=l&&y>=r)
    {
        vec tmp=u;
        int t=0;
        pushup(t,u,tmp,sgt[p].ve);
        return t;
    }
    int mid=(l+r)/2,res=0;
    if(x<=mid)
        res=query(p*2,x,y,u);
    if(y>mid)
        res=query(p*2+1,x,y,u);
    return res;//用两个儿子更新爸爸
}
int cnt[205];
int main()
{
    cin>>n>>m>>q;
    for(int i=1;i<=m;i++)
        cin>>e[i].a>>e[i].b>>e[i].w;
    iniSgt(1,1,m);
    while(q--)
    {
        int opt;cin>>opt;
        if(opt==1)
        {
            int x,y,z,t;cin>>x>>y>>z>>t;
            e[x].a=y,e[x].b=z,e[x].w=t;
            modify(1,x);
        }
        else
        {
            int l,r;cin>>l>>r;
            vec t;int tot=0;
            int ans=query(1,l,r,t);
            if(t.size()==n-1)cout<<ans<<endl;
            else cout<<"Impossilbe"<<endl;
        }
    }
}