Codeforces 1296 F Berland Beauty —— 树链剖分,lca

This way

题意:

给你一棵树,每条边都有一个权值,告诉你m条路径上的权值的最小值,让你反推每条边的权值的任意一种情况。

题解:

n 2 n^2 也能做啊,但是标准应该是树链剖分。权值存边变为存点,存到儿子上。然后其实就只需要将告诉你的m个路径,更新最大值即可,因为它告诉你的是这条路径上的最小值,那么这条路径上是否可能还有更大的值存在而不可能有更小的值存在。做完之后再做一遍查看每条路径上的最小值是否就是输入的那样即可,然后对于其它未知点,我们默认是1e6即可。
但是你看看这是人干的事情吗
在这里插入图片描述
我等了半天它一个一个给我跑过去,我****

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pa pair<int,int>
const int N=1e4+5;
const int inf=1e9;
struct node
{
    int to,next;
}e[N*2];
int head[N],cnt,tim,fa[N][21],son[N],dep[N],top[N],en[N],siz[N],pos[N];
int flag[N*4],sum[N*4],mi[N*4];
int n,m,rt;
void add(int x,int y)
{
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
void dfs1(int u)
{
    siz[u]=1;
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u][0])
        {
            fa[v][0]=u;
            dep[v]=dep[u]+1;
            dfs1(v);
            siz[u]+=siz[v];
            if (son[u]==0||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}
void dfs2(int u,int f)
{
    tim++;
    top[u]=f;
    pos[u]=tim;
    if (son[u]) dfs2(son[u],f);
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u][0]&&v!=son[u])
        {
            dfs2(v,v);
        }
    }
    en[u]=tim;
}
void pushdown(int root)
{
    if(!flag[root])return ;
    sum[root<<1]=max(sum[root<<1],flag[root]);
    sum[root<<1|1]=max(sum[root<<1|1],flag[root]);
    //mi[root<<1]=mi[root<<1|1]=flag[root];
    flag[root<<1]=max(flag[root<<1],flag[root]);
    flag[root<<1|1]=max(flag[root<<1|1],flag[root]);
    flag[root]=0;
}
void add(int l,int r,int root,int ql,int qr,int val)
{
    if(l>=ql&&r<=qr)
    {
        sum[root]=max(sum[root],val);
        flag[root]=max(flag[root],val);
        //mi[root]=val;
        return ;
    }
    pushdown(root);
    int mid=l+r>>1;
    if(mid>=ql)
        add(l,mid,root<<1,ql,qr,val);
    if(mid<qr)
        add(mid+1,r,root<<1|1,ql,qr,val);
    //pushup(root);
}
void build(int l,int r,int root){
    if(l==r){
        mi[root]=sum[root]?sum[root]:1e6;
        return ;
    }
    pushdown(root);
    int mid=l+r>>1;
    build(l,mid,root<<1);
    build(mid+1,r,root<<1|1);
    mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
int ask(int l,int r,int root,int ql,int qr)
{
    if(l>=ql&&r<=qr)
        return mi[root];
    int mid=l+r>>1;
    pushdown(root);
    int ans=inf;
    if(mid>=ql)
        ans=ask(l,mid,root<<1,ql,qr);
    if(mid<qr)
        ans=min(ask(mid+1,r,root<<1|1,ql,qr),ans);
    return ans;
}
int query(int x,int y)
{
    int ans=inf;
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=min(ans,ask(1,n,1,pos[top[x]],pos[x]));
        x=fa[top[x]][0];
    }
    if (pos[x]>pos[y]) swap(x,y);
    ans=min(ans,ask(1,n,1,pos[x],pos[y]));
    return ans;
}
void update(int x,int y,int val)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        add(1,n,1,pos[top[x]],pos[x],val);
        x=fa[top[x]][0];
    }
    if (pos[x]>pos[y]) swap(x,y);
    add(1,n,1,pos[x],pos[y],val);
}
void deal(){
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(int i=15;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y)return x;
    for(int i=15;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
struct value{
    int x,y,v;
}va[N];
struct edge{
    int x,y;
}edg[N];
int ans[N];
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    int x,y,v;
    for(int i=1;i<n;i++)
        scanf("%d%d",&edg[i].x,&edg[i].y),add(edg[i].x,edg[i].y),add(edg[i].y,edg[i].x);
    dep[1]=1;
    dfs1(1);
    dfs2(1,1);
    deal();
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&v);
        va[i]={x,y,v};
        int l=lca(x,y),firstson;
        if(y!=l)
        {
            firstson=y;
            for(int i=15;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            update(y,firstson,v);
        }
        if(x!=l)
        {
            firstson=x;
            for(int i=15;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            update(x,firstson,v);
        }
    }
    build(1,n,1);
    for(int i=1;i<=m;i++){
        x=va[i].x,y=va[i].y,v=va[i].v;
        int l=lca(x,y),firstson;
        int a=inf;
        if(y!=l)
        {
            firstson=y;
            for(int i=15;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            a=query(y,firstson);
        }
        if(x!=l)
        {
            firstson=x;
            for(int i=15;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            a=min(a,query(x,firstson));
        }
        if(a!=v)
            return 0*printf("-1\n");
    }
    for(int i=1;i<n;i++){
        int p=dep[edg[i].x]>dep[edg[i].y]?pos[edg[i].x]:pos[edg[i].y];
        v=ask(1,n,1,p,p);
        printf("%d\n",v);
    }
    return 0;
}

发布了584 篇原创文章 · 获赞 33 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/104659979