[最小生成树 并查集] Codechef October Challenge 2017 .Lucky Edge

O(n) 枚举左端点,然后往右做一棵以编号为权值的最小生成树,每一条非树边覆盖的树边的编号一定小于它。

那么再从左往右扫每条非树边,会对它覆盖的树边加上 ni+1 的贡献,但是树边只能被加一次

所以可以用并查集把加过的树边的两个端点缩起来。

这样也就不需要求LCA了

O(n2α)

然后卡卡常就好了…

#include 
#include 
#include 

using namespace std;

const int N=10010;

int n,cnt,p[N],fa[N],dpt[N],frm[N],a[N],b[N];
int t,l[N],ans[N],vis[N],it[N];
struct edge{
    int t,fr;
    edge *nxt;
}E[N<<1],*G[N];

inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void rea(int &x){
    char c=nc(); x=0;
    for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}

inline void add(const int &x,const int &y,const int &z){
    E[++cnt].t=y; E[cnt].nxt=G[x]; E[cnt].fr=z; G[x]=E+cnt;
    E[++cnt].t=x; E[cnt].nxt=G[y]; E[cnt].fr=z; G[y]=E+cnt;
}

int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}

void dfs(int x,int f,int fr){
    p[x]=f; frm[x]=fr; vis[x]=1; dpt[x]=dpt[f]+1;
    for(edge *i=G[x];i;i=i->nxt)
        if(i->t!=f) dfs(i->t,x,i->fr);
}

int fa1[N];

int find1(int x){
    return x==fa1[x]?x:fa1[x]=find1(fa1[x]);
}

int v[N],vt;

int main(){
    int T; rea(T);
    while(T--){
        rea(n); t=0;
        for(int i=1;i<=n;i++)
            rea(a[i]),rea(b[i]),l[++t]=a[i],l[++t]=b[i],ans[i]=0;
        sort(l+1,l+1+t); t=unique(l+1,l+1+t)-l-1;
        for(int i=1;i<=n;i++)
            a[i]=lower_bound(l+1,l+1+t,a[i])-l,b[i]=lower_bound(l+1,l+1+t,b[i])-l;
        for(int l=1;l<=n;l++){
            for(int i=1;i<=t;i++) fa[i]=fa1[i]=i,G[i]=0,vis[i]=0;
            cnt=vt=0;
            for(int i=l;i<=n;i++)
                if(find(a[i])!=find(b[i])){
                    add(a[i],b[i],i); fa[find(a[i])]=b[i];
                }
                else v[++vt]=i;
            if(!vt) continue;
            for(int i=1;i<=vt;i++){
                int u=v[i]; ans[u]+=n-u+1;
                int x=a[u],y=b[u]; x=find1(x); y=find1(y);
                if(!vis[x]) dfs(x,0,0);
                while(x!=y){
                    if(dpt[x]1; fa1[x]=p[x];
                    x=find1(x);
                }
            }
        }
        for(int i=1;i<=n;i++) printf("%d ",ans[i]); putchar('\n');
    }
    return 0;
} 

你可能感兴趣的:(最小-最大生成树,并查集)