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

来源:互联网 发布:网络词ip是什么意思 编辑:程序博客网 时间:2024/06/05 19:24

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

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

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

这样也就不需要求LCA了

O(n2α)

然后卡卡常就好了…

#include <cstdio>#include <iostream>#include <algorithm>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]<dpt[y]) swap(x,y);                    ans[frm[x]]+=n-u+1; fa1[x]=p[x];                    x=find1(x);                }            }        }        for(int i=1;i<=n;i++) printf("%d ",ans[i]); putchar('\n');    }    return 0;} 
原创粉丝点击