入门OJ:八中生成树2

题目描述

八中里面有N个建设物，M条边。对于这种要建最小生成树的问题，你应该很熟练了。现在老大决定降低某条边的费用，然后这条边必须要被选中，因为这条路他每天都要走，自然......问选了这条边后是否可以得到一个比从前总开支相等或更小的方案。

输入格式

第一行三个整数N,M,Q(1<=N<=1000,N-1<=M<=100000接下来M行，每行三个整数（X，Y,Z）描述一条可以建造的路。 0<=Z<=10000 最后Q行，每行两个整数i,x(1<=i<=M,0<=x)描述一个修改计划，即要你判断如果 将第i条公路的修建费用降低为x元后，是否可以得到一个比从前总开支相等或更小的方案

输出格式

按顺序输出Q行，每行输出"Yes"或"No",Yes表示可以建造，No表示不可能

---

先考虑暴力。

我们可以每次询问之后都修改一次边权，然后在原图求一次最小生成树，看是否更优即可。时间复杂度为O(Q * MlogM)

其实在原图再求一次mst时基本上所有边都是不需要变动的。可以利用这一点考虑优化

如果询问给出的边是树上边，那么只需比较和原来的树边大小即可。

如果不是树上边，那么如果原树不变，只加上询问的边，就会形成一个环。设询问给出的边为E(u,v)，那么这个环就可以表示为：u->LCA(u,v)->v->u。所以生成树上唯一需要改变的就是这个环上的部分。只需要把环断掉一条边即可。为了使断掉之后的生成树尽可能小，我们肯定会断掉环上最长的边(除了询问给的边)。所以问题就变成了：求路径u->LCA(u,v)->v上最长边。

可以用倍增做，时间复杂度为O((N+M)logN)，总复杂度就是：O((N+Q)logN+MlogM)。也可以树剖直接做，复杂度为O(QlogNlogN+MlogM)。

#include
#include
#include
#include
#define maxn 1001
#define maxm 100001
using namespace std;
 
struct brim{
    int u,v,w;
    bool operator<(const brim &b)const{ return wsize[son[u]]) son[u]=v;
    }
}
void dfs_rewrite(int u,int tp){
    top[u]=tp,dfn[u]=++tot,id[tot]=u;
    if(son[u]) dfs_rewrite(son[u],tp);
    for(register int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(v!=son[u]&&v!=fa[u]) dfs_rewrite(v,v);
    }
}
inline void endow(){
    for(register int i=0;idep[v]) swap(u,v);
        val[v]=e[i].dis;
    }
}
 
void build(int d,int l,int r){
    t[d].l=l,t[d].r=r;
    if(l==r){ t[d].c=val[id[l]]; return; }
    int mid=l+r>>1;
    build(d<<1,l,mid),build(d<<1|1,mid+1,r);
    t[d].c=max(t[d<<1].c,t[d<<1|1].c);
}
int getmax(int d,const int &l,const int &r){
    if(l<=t[d].l&&t[d].r<=r) return t[d].c;
    int mid=t[d].l+t[d].r>>1,ans=0;
    if(l<=mid) ans=max(ans,getmax(d<<1,l,r));
    if(r>mid) ans=max(ans,getmax(d<<1|1,l,r));
    return ans;
}
 
int main(){
    memset(head,-1,sizeof head);
    n=read(),m=read(),q=read();
    for(register int i=1;i<=m;i++) b[i].u=read(),b[i].v=read(),b[i].w=read();
    kruskal();
 
    dfs_getson(1);
    dfs_rewrite(1,1);
    endow();
    build(1,1,tot);
 
    for(register int i=1;i<=q;i++){
        int id=read(),w=read(),mmax=0,u=b[id].u,v=b[id].v;
        while(top[u]!=top[v]){
            if(dep[top[u]]>dep[top[v]]) swap(u,v);
            mmax=max(mmax,getmax(1,dfn[top[v]],dfn[v]));
            v=fa[top[v]];
        }
        if(dep[u]>dep[v]) swap(u,v);
        if(u!=v) mmax=max(mmax,getmax(1,dfn[u]+1,dfn[v]));
        printf("%s\n",mmax-w>=0?"Yes":"No");
    }
    return 0;
}

转载于:https://www.cnblogs.com/akura/p/10885980.html