Live2d Test Env

POJ - 3728:The merchant (Tarjan 带权并查集)

题意:给定一个N个节点的树,1<=N<=50000 每个节点都有一个权值,代表商品在这个节点的价格。商人从某个节点a移动到节点b,且只能购买并出售一次商品,问最多可以产生多大的利润。

思路:路径压缩,得到每个点到当前根的信息,然后更新即可。 有可以用倍增做。

很久前抄的代码。

#include<cstdio>
#define min(a,b) (a<b?a:b)
#define max(a,b) (a>b?a:b)
#define swap(a,b) (a^=b,b^=a,a^=b)
using namespace std;
const int mm=333333;
const int mn=55555;
int s[mm],t[mm],d[mm],p[mm],ans[mm];
int h[mn]={0},q[mn]={0},g[mn]={0},f[mn],mx[mn],mi[mn],up[mn]={0},dw[mn]={0};
bool vis[mn]={0};
int i,j,k,n,m,e;
inline void add(int u,int v,int c,int h[])
{
    s[e]=u,t[e]=v,d[e]=c,p[e]=h[u],h[u]=e++;
    s[e]=v,t[e]=u,d[e]=-c,p[e]=h[v],h[v]=e++;
}
int find(int x)
{
    if(f[x]==x)return x;
    int y=f[x];
    f[x]=find(f[x]);
    up[x]=max(mx[y]-mi[x],max(up[x],up[y]));
    dw[x]=max(mx[x]-mi[y],max(dw[x],dw[y]));
    mx[x]=max(mx[x],mx[y]);
    mi[x]=min(mi[x],mi[y]);
    return f[x];
}
void tarjan(int u)
{
    int i,v,x,y;
    vis[f[u]=u]=1;
    for(i=q[u];i;i=p[i])
        if(vis[v=t[i]])
            v=find(v),t[e]=i,p[e]=g[v],g[v]=e++;
    for(i=h[u];i;i=p[i])
        if(!vis[v=t[i]])tarjan(v),f[v]=u;
    for(i=g[u];i;i=p[i])
    {
        v=t[i],x=s[v],y=t[v],find(x);
        if(d[v]<0)swap(x,y),v=-d[v];
        else v=d[v];
        ans[v]=max(mx[y]-mi[x],max(up[x],dw[y]));
    }
}
inline void get(int &a)
{
    char c;
    while((c=getchar())<'0'||c>'9');
    for(a=0;c>='0'&&c<='9';c=getchar())a=a*10+c-'0';
}
void out(int x)
{
    if(x>9)out(x/10);
    putchar(x%10+'0');
}
int main()
{
    for(get(n),i=1; i<=n; ++i)get(mx[i]),mi[i]=mx[i];
    for(e=k=1; k<n; ++k)get(i),get(j),add(i,j,0,h);
    for(get(m),k=1; k<=m; ++k)get(i),get(j),add(i,j,k,q);
    tarjan(1);
    for(i=1; i<=m; ++i)out(ans[i]),puts("");
    return 0;
}

 

posted @ 2019-07-24 11:57  nimphy  阅读(241)  评论(0编辑  收藏  举报