3712: [PA2014]Fiolki LCA 思路题

非常神的一道题。。orzzzzz
做法:
我们用一个节点代表一个瓶子,合并的时候新建一个节点,代表合并出来的瓶子,并且向原来的两个点连边。可以发现,反应的话,其实就是两个瓶子代表节点的LCA。
然后按照LCA的深度以及时间戳排序,算答案就好了。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

const int N=400005;
const int Q=500005;
int head[N],next[N],list[N];
int n,m,k,top,cnt,dfn;
long long ans;
int fa[N][19],deep[N],belong[N],g[N],pos[N];

struct C
{
    int x,y,d,id;
    C () {};
    C (int _,int __,int ___,int ____)
    {
        x=_; y=__; d=___; id=____;
    }
}q[Q];

inline bool operator<(C a,C b)
{
    return a.d==b.d?a.id<b.id:a.d>b.d;
}

inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}

inline void insert(int x,int y)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
}

void dfs(int x)
{
    belong[x]=dfn;
    for (int i=1;(1<<i)<=deep[x];i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for (int i=head[x];i;i=next[i])
        if (list[i]!=fa[x][0])
        {
            fa[list[i]][0]=x;
            deep[list[i]]=deep[x]+1;
            dfs(list[i]);
        }
}

inline int lca(int x,int y)
{
    if (deep[x]<deep[y]) swap(x,y);
    int t=deep[x]-deep[y];
    for (int i=0;(1<<i)<=t;i++)
        if ((1<<i)&t) x=fa[x][i];
    for (int i=18;~i;i--)
        if (fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}

int main()
{
    n=read(); m=read(); k=read();
    for (int i=1;i<=n;i++) g[i]=read();
    for (int i=1;i<=n;i++) pos[i]=i;
    for (int i=1;i<=m;i++)
    {
        int u=read(),v=read(),f=n+i;
        insert(f,pos[u]); insert(f,pos[v]);
        pos[v]=f;
    }
    for (int i=n+m;i;i--)
        if (!fa[i][0]) dfn++,dfs(i);
    for (int i=1;i<=k;i++)
    {
        int u=read(),v=read();
        if (belong[u]!=belong[v]) continue;
        int f=lca(u,v);
        q[++top]=C(u,v,deep[f],i);
    }
    sort(q+1,q+top+1);
    for (int i=1;i<=top;i++)
    {
        int x=q[i].x,y=q[i].y;
        int mn=min(g[x],g[y]);
        g[x]-=mn; g[y]-=mn; ans+=mn<<1;
    }
    cout << ans << endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值