其他分享
首页 > 其他分享> > test20190504 行走

test20190504 行走

作者:互联网

行走(walk.cpp/c/pas)

题目描述

“我有个愿望,我希望走到你身边。”
这是个奇异的世界,世界上的 n-1 条路联结起来形成一棵树,每条路有一个对应的权值 ci。
现在我会给出 q 组询问或操作。
每次询问我会从一个 x 点走到 y 点,初始在 x 点我会有一个数字 v,然后每走过一条权值为 c 的边,我的 v
就会变成\(\lfloor \frac vc \rfloor\) ,问最后到 y 时 v 变成了什么。
每次修改我会修改一条边的权值,保证修改后的权值小于等于原来的权值且不会小于 1。
每组询问或操作的格式如下:
询问:1 x y v 表示从 x 走到 y,一开始的数字为 v。
操作:2 p c 表示将第 p 条边的边权修改为 c

输入

第一行两个整数 n 和 q 表示点个数和询问与操作个数
接下来 n-1 行每行三个整数 u,v,c 表示 u 与 v 之间有一条边权为 c 的边
接下来 q 行每行第一个数 type
如果 type=1 那么接下来三个数 x,y,v 表示一组询问
如果 type=2 那么接下来两个数 p,c 表示一组操作

输出

对于每组询问输出一个数表示最后的答案

样例输入 1

6 6
1 2 1
1 3 7
1 4 4
2 5 5
2 6 2
1 4 6 17
2 3 2
1 4 6 17
1 5 5 20
2 4 1
1 5 1 3

样例输出 1

2
4
20
3

样例输入 2

5 4
1 2 7
1 3 3
3 4 2
3 5 5
1 4 2 100
1 5 4 1
2 2 2
1 1 3 4

样例输出 2

2
0
2

数据范围

对于 70%的数据保证\(1\le n\le 2000\)
对于 100%的数据保证\(1\le n \le 100000,1\le c_i\le 10^{18}\)
保证每次修改后的边权小于等于原来的边权且不会小于 1

分析

考场做法

如果走过的边权不是1,每次至少除2,只有\(log_2n\)次有效操作。直觉想法是把边权为1的边连接的点用并查集连起来,缩成一个等效点。

但这样做有一个问题,那就是向上跳可以直接跳fa,向下跳呢?于是我想出来用set维护边,给边附加一个maxdfn信息,然后二分查找,这样就不会被菊花图卡了。

然后觉得代码不止一点恶心,颓了半小时,终于狠下心来写完了,然后改了一会过了样例,简直不敢相信自己的眼睛。

时间复杂度\(O(n\log n+q\log v\log n)\)

#include<bits/stdc++.h>
#define co const
template<class T>T read(){
    T data=0,w=1;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') w=-1;ch=getchar();}
    while(isdigit(ch)) data=data*10+ch-'0',ch=getchar();
    return data*w;
}
template<class T>T read(T&x){
    return x=read<T>();
}
typedef long long ll;
using namespace std;

co int N=1e5+1;
int n,m;
struct org_edge{int x,y;ll c;}oe[N]; // real
struct edge{ // vitual
    int mxp,to;ll c;
    bool operator<(co edge&e)co{
        return mxp!=e.mxp?mxp<e.mxp:to<e.to;
    }
};
set<edge> e[N];
typedef set<edge>::iterator it;

int dep[N],pos[N],dfn,fa[N][18];ll val[N];
void dfs(int x,int fa){ // real
    dep[x]=dep[fa]+1,pos[x]=++dfn,::fa[x][0]=fa;
    for(int i=1;i<=17;++i) ::fa[x][i]=::fa[::fa[x][i-1]][i-1];
    set<edge> s;
    for(it i=e[x].begin();i!=e[x].end();++i){
        int y=i->to;
        if(y==fa) {val[x]=i->c;continue;}
        dfs(y,x),s.insert((edge){dfn,y,i->c});
    }
    swap(e[x],s);
}
int lca(int x,int y){ // real
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=17;i>=0;--i)
        if(dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if(x==y) return x;
    for(int i=17;i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

int pa[N];
int find(int x) {return x==pa[x]?x:pa[x]=find(pa[x]);}
void link(int x,int y){ // virtual
    assert(x==find(x)),assert(y==find(::fa[x][0]));
    pa[x]=y;
    it i=e[y].lower_bound((edge){pos[x],0,0});
    assert(pos[x]<=i->mxp),e[y].erase(i);
    for(i=e[x].begin();i!=e[x].end();++i) e[y].insert(*i);
    e[x].clear();
}

ll query(int x,int y,ll v){ // real->virtual
    int f=lca(x,y);
    x=find(x),y=find(y),f=find(f);
    for(;v&&x!=f;x=find(::fa[x][0])) v/=val[x];
    if(!v) return 0;
    assert(x==f);
    for(it i;v&&x!=y;x=i->to){
        i=e[x].lower_bound((edge){pos[y],0,0});
        v/=i->c;
    }
    return v;
}
void change(int p,ll c){ // real->virtual
    int x=find(oe[p].x),y=find(oe[p].y);
    if(x==y) return assert(c==1);
    if(dep[x]>dep[y]) swap(x,y);
    assert(x==find(::fa[y][0]));
    it i=e[x].lower_bound((edge){pos[y],0,0});
    assert(pos[y]<=i->mxp),e[x].erase(i),e[x].insert((edge){i->mxp,i->to,c}),val[y]=c;
    if(c==1) link(y,x);
}
int main(){
    freopen("walk.in","r",stdin),freopen("walk.out","w",stdout);
    read(n),read(m);
    for(int i=1;i<n;++i){
        read(oe[i].x),read(oe[i].y),read(oe[i].c);
        e[oe[i].x].insert((edge){0,oe[i].y,oe[i].c});
        e[oe[i].y].insert((edge){0,oe[i].x,oe[i].c});
    }
    dfs(1,0);
    for(int i=1;i<=n;++i) pa[i]=i;
    for(int i=1;i<=n;++i)if(val[i]==1) link(i,find(::fa[i][0]));
    while(m--){
        if(read<int>()==1){
            int x,y;ll v;
            read(x),read(y),read(v);
            printf("%lld\n",query(x,y,v));
        }
        else{
            int p;ll c;
            read(p),read(c);
            change(p,c);
        }
    }
    return 0;
}

刘哥做法

打表得到性质,下取整除法可以结合(意会),并且要证明的话是显然的。

于是树剖线段树维护路径乘积,维护一下溢出标记,然后每次把路径找出来就行了。

时间复杂度\(O(n+m\log^n)\)

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
inline ll read()
{
    ll x=0,k=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-') k=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return k*x;
}
const int MAXN=2e5+10;
int n,Q;
int ecnt=0,head[MAXN],to[MAXN<<1],nx[MAXN<<1],fa[MAXN];
ll val[MAXN<<1],vf[MAXN];
int dep[MAXN];
int U[MAXN],V[MAXN];
inline void addedge(int u,int v,ll w)
{
    ++ecnt;
    to[ecnt]=v;
    nx[ecnt]=head[u];
    val[ecnt]=w;
    head[u]=ecnt;
}
ll wp[MAXN];
int dfn[MAXN],dfnidx=0,rnk[MAXN],siz[MAXN],mxson[MAXN],top[MAXN];
void dfs1(int u,int f)
{
    fa[u]=f;
    siz[u]=1;
    dep[u]=dep[f]+1;
    for(int i=head[u]; i; i=nx[i])
    {
        int v=to[i];
        if(v==f)
            continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        if(siz[v]>siz[mxson[u]])
            mxson[u]=v;
    }
}
void dfs2(int u,int tp)
{
    dfn[u]=++dfnidx;
    rnk[dfnidx]=u;
    top[u]=tp;
    if(mxson[u])
        dfs2(mxson[u],tp);
    for(int i=head[u]; i; i=nx[i])
    {
        int v=to[i];
        if(v!=mxson[u] && v!=fa[u])
            dfs2(v,v);
    }
}
ll prod[MAXN<<2];
#define root prod[o]
#define lson prod[o<<1]
#define rson prod[o<<1|1]
void pushup(int o)
{
    root=lson*rson;
    root=max(root,0LL);
}
void bd(int o,int l,int r)
{
    if(l==r)
    {
        root=wp[rnk[l]];
        return;
    }
    int mid=(l+r)>>1;
    bd(o<<1,l,mid);
    bd(o<<1|1,mid+1,r);
    pushup(o);
}
void update(int o,int l,int r,int pos,ll c)
{
    if(l==r)
    {
        root=c;
        return;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
        update(o<<1,l,mid,pos,c);
    else
        update(o<<1|1,mid+1,r,pos,c);
    pushup(o);
}
ll query(int o,int l,int r,int L,int R)
{
    ll res=1;
    if(l>R || L>r)
        return 1;
    if(L<=l && r<=R)
        return max(0LL,root);
    int mid=(l+r)>>1;
    if(L<=mid)
    {
        res*=query(o<<1,l,mid,L,R);
        res=max(res,0LL);
    }
    if(R>mid)
    {
        res*=query(o<<1|1,mid+1,r,L,R);
        res=max(res,0LL);
    }
    return res;
}
void solve()
{
    int idx=n;
    for(int i=1; i<n; ++i)
    {
        int u=read(),v=read();
        ll w=read();
        ++idx;
        addedge(u,idx,0);
        addedge(idx,u,0);
        addedge(idx,v,0);
        addedge(v,idx,0);
        wp[idx]=w;
    }
    for(int i=1; i<=n; ++i)
        wp[i]=1;
    dfs1(1,0);
    dfs2(1,1);
    bd(1,1,idx);
    while(Q--)
    {
        int tp=read();
        if(tp==1)
        {
            int x=read(),y=read();
            ll v=read();
            while(top[x]!=top[y])
            {
                if(dep[top[x]]<dep[top[y]])
                    swap(x,y);
                ll p=query(1,1,idx,dfn[top[x]],dfn[x]);
                if(p<=0)
                {
                    puts("0");
                    continue;
                }
                v/=p;
                x=fa[top[x]];
            }
            if(dep[x]<dep[y])
                swap(x,y);
            ll p=query(1,1,idx,dfn[y],dfn[x]);
            if(p<=0)
            {
                puts("0");
                continue;
            }
            v/=p;
            printf("%lld\n",v);
        }
        else
        {
            int p=read();
            ll c=read();
            update(1,1,idx,dfn[p+n],c);
        }
    }
}
void dfs(int u,int f)
{
    fa[u]=f;
    for(int i=head[u]; i; i=nx[i])
    {
        int v=to[i];
        if(v==f)
            continue;
        dep[v]=dep[u]+1;
        vf[v]=val[i];
        dfs(v,u);
    }
}
int bf()
{
    for(int i=1; i<n; ++i)
    {
        int u=read(),v=read();
        ll w=read();
        U[i]=u,V[i]=v;
        addedge(u,v,w);
        addedge(v,u,w);
    }
    dfs(1,0);
    while(Q--)
    {
        int tp=read();
        if(tp==1)
        {
            int x=read(),y=read();
            ll v=read();
            while(x!=y)
            {
                if(dep[x]>dep[y])
                    v/=vf[x],x=fa[x];
                else
                    v/=vf[y],y=fa[y];
            }
            printf("%lld\n",v);
        }
        else
        {
            int p=read();
            ll v=read();
            if(fa[U[p]]==V[p])
                vf[U[p]]=v;
            else
                vf[V[p]]=v;
        }
    }
    return 0;
}
int main()
{
    freopen("walk.in","r",stdin);
    freopen("walk.out","w",stdout);
    n=read();
    Q=read();
    if(n<=1000)
        return bf();
    solve();
    return 0;
}

std做法

其实往下跳没有必要了。过了lca以后仍然从另一边往上跳,溢出及时跳出就行了。这样也最多\(\log_2 v\)次

标程暂缺。陈年老题,估计也不会有了。

放上前辈的代码。

//QWsin
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=100000+10;
const int maxm=200000+10;

int n,Q;

int first[maxn],next[maxm],ecnt=0;
struct Edge{int u,v;ll w;Edge(int u=0,int v=0,ll w=0):u(u),v(v),w(w){}}e[maxm];
inline void add_edge(int u,int v,ll w)
{
    next[ecnt]=first[u];first[u]=ecnt;e[ecnt++]=Edge(u,v,w);
    next[ecnt]=first[v];first[v]=ecnt;e[ecnt++]=Edge(v,u,w);
}

int dep[maxn],fa[maxn],faid[maxn];
void dfs(int u,int pre,int deep)
{
    dep[u]=deep;
    for(int i=first[u];i!=-1;i=next[i])
        if(e[i].v!=pre) 
        {
            fa[e[i].v]=u;faid[e[i].v]=i;
            dfs(e[i].v,u,deep+1);   
        }
}

ll query(int a,int b,ll c)
{
    if(dep[a] > dep[b]) swap(a,b);
    if(dep[a] < dep[b]){
        for(;dep[a]<dep[b];b=fa[b])
            if((c/=e[faid[b]].w)==0) return 0;
    }
    if(a!=b)
    {
        for(;a!=b;a=fa[a],b=fa[b]){
            if((c/=e[faid[b]].w)==0) return 0;
            if((c/=e[faid[a]].w)==0) return 0;
        }
    }
    return c;
}

int main()
{
    freopen("walk.in","r",stdin);
    freopen("walk.out","w",stdout);
    cin>>n>>Q;ll w;
    memset(first,-1,sizeof first);
    for(int i=1,u,v;i<n;++i) {
        scanf("%d%d%lld",&u,&v,&w);add_edge(u,v,w);
    }
    
    dfs(1,1,1);
    
    int op,a,b;ll c;
    while(Q--)
    {
        scanf("%d",&op);
        if(op==1) {
            scanf("%d%d%lld",&a,&b,&c);
            printf("%lld\n",query(a,b,c));
        }
        else{
            scanf("%d%lld",&a,&c);
            e[a*2-1].w=e[a*2-2].w=c;
        }
    }
    return 0;
}

标签:dep,ch,fa,int,test20190504,ll,read,行走
来源: https://www.cnblogs.com/autoint/p/10808421.html