[PA2014]Fiolki 解题报告

这道题是delayyy拿来当例题给我们讲的,他讲得启发式合并,以外向边与点数的和为关键字(因为这是我们一次合并需要的代价),可以做到 O(m+(n+k)log(n+k))
但是。。显然这是一棵树嘛,一棵树的话你直接求LCA就好了,求LCA可以Tarjan做到 O(n) ,从哪棵子树上来的可以在ufs里维护一下,然后把所有的点对按照询问给出的顺序插入就可以代替排序的效果了。所以这样可以做到 O(n+m+k) ,所以就很愉快的刷了Rank1!
一个容易出问题的地方是并查集维护的时候要先压缩。。再维护信息。于是我很愉快地RE了20遍。。
= =因为是权限题所以只能找学长借的号。

#include
#include
#include
#include
#include
using namespace std;
char * c=(char *)malloc(30000000);
inline void in(int &x){
    x=0;
    while(*c<'0'||*c>'9')++c;
    while(*c>='0'&&*c<='9')x=x*10+(*c++^'0');
}
int g[200005];
#include
int sonptr[200005],sonsize[200005],sonnext[200005],sonsucc[200005];
bool p[200005];
int qptr[200005],qnext[1000005],ab[1000005];
int lca[500005],bson[500005],fa[200005],wson[200005],wsn[200005];
int find(int x){
    if(fa[x]!=fa[fa[x]]){
        int ftr=fa[x];
        fa[x]=find(fa[x]);
        wson[x]=wson[ftr];
    }
    return fa[x];
}
void tarjan(int node,int ftr){
    fa[node]=node;
    for(int i=sonptr[node],j=0;i;i=sonnext[i],++j){
        wsn[node]=j;
        tarjan(sonsucc[i],node);
        wson[sonsucc[i]]=j;
    }
    for(int i=qptr[node];i;i=qnext[i])
        if(p[node^ab[i]]){
            lca[i-1>>1]=find(node^ab[i]);
            if(lca[i-1>>1]==node)bson[i-1>>1]=wson[node^ab[i]];
            else bson[i-1>>1]=wsn[lca[i-1>>1]];
        }
    p[node]=1;
    fa[node]=ftr;
}

int a[500005],b[500005];
int ptr[200005],sons[200005];
int next[500005],succ[500005];
long long ans;
void dfs(int node){
    int tmp;
    for(int i=sonptr[node],k=0,j;i;i=sonnext[i],++k){
        dfs(sonsucc[i]);
        for(j=ptr[sons[node]+k];j;j=next[j]){
            tmp=min(g[a[succ[j]]],g[b[succ[j]]]);
            g[a[succ[j]]]-=tmp,g[b[succ[j]]]-=tmp;
            ans+=tmp<<1;
            //if(tmp)cout<<"Get("<int main(){
    freopen("fiolki.in","r",stdin);
    freopen("fiolki.out","w",stdout);
    int n,m,k;
    fread(c,1,30000000,stdin);
    in(n),in(m),in(k);
    for(int i=1;i<=n;++i)in(g[i]);
    int stot=1,rtot=1,qtot=1;
    for(int i=0;ifor(int i=m;i--;){
        ++sonsize[b[i]];
        sonnext[stot]=sonptr[b[i]];
        sonptr[b[i]]=stot;
        sonsucc[stot++]=a[i];
        p[a[i]]=1;
    }
    for(int i=n;i;--i)
        if(!p[i]){
            ++sonsize[0];
            sonnext[stot]=sonptr[0];
            sonptr[0]=stot;
            sonsucc[stot++]=i;
        }
    for(int i=0;imemset(p,0,sizeof(p));
    for(int i=sonptr[0];i;i=sonnext[i])tarjan(sonsucc[i],0);
    sons[1]=1;
    for(int i=2;i<=n;++i)sons[i]=sons[i-1]+sonsize[i-1];
    for(int i=k;i--;)
        if(lca[i]){
            next[rtot]=ptr[sons[lca[i]]+bson[i]];
            succ[rtot]=i;
            ptr[sons[lca[i]]+bson[i]]=rtot++;
            //cout<
        }
    for(int i=sonptr[0];i;i=sonnext[i])dfs(sonsucc[i]);
    printf("%lld\n",ans);
}

你可能感兴趣的:(UFS)