[UOJ261]天天爱跑步

问题描述:小C同学认为跑步非常有趣,于是决定制作一款叫做《天天爱跑步》的游戏。《天天爱跑步》是一个养成类游戏,需要玩家每天按时上线,完成打卡任务。

这个游戏的地图可以看作一棵包含 n 个结点和 n−1 条边的树,每条边连接两个结点,且任意两个结点存在一条路径互相可达。树上结点编号为从 1 到 n 的连续正整数。

现在有 mm 个玩家,第 ii 个玩家的起点为 Si,终点为 Ti。每天打卡任务开始时,所有玩家在第 00 秒同时从自己的起点出发,以每秒跑一条边的速度,不间断地沿着最短路径向着自己的终点跑去,跑到终点后该玩家就算完成了打卡任务。(由于地图是一棵树,所以每个人的路径是唯一的)

小C想知道游戏的活跃度,所以在每个结点上都放置了一个观察员。在结点 j 的观察员会选择在第 Wj 秒观察玩家,一个玩家能被这个观察员观察到当且仅当该玩家在第 Wj 秒也正好到达了结点 j。小C想知道每个观察员会观察到多少人?

注意:我们认为一个玩家到达自己的终点后该玩家就会结束游戏,他不能等待一段时间后再被观察员观察到。即对于把结点 j 作为终点的玩家:若他在第 Wj 秒前到达终点,则在结点 j 的观察员不能观察到该玩家;若他正好在第 Wj 秒到达终点,则在结点 j 的观察员可以观察到这个玩家。

输入格式
从标准输入读入数据。

第一行有两个整数 n 和 m。其中 n 代表树的结点数量,同时也是观察员的数量,m 代表玩家的数量。

接下来 n−1 行每行两个整数 u 和 v,表示结点 u 到结点 v 有一条边。

接下来一行 n 个整数,其中第 j 个整数为 Wj,表示结点 j 出现观察员的时间。

接下来 mm 行,每行两个整数 Si 和 Ti,表示一个玩家的起点和终点。

对于所有的数据,保证 1≤Si,Ti≤n1≤Si,Ti≤n,0≤Wj≤n0≤Wj≤n。

输出格式
输出到标准输出。

输出 11 行 n 个整数,第 j 个整数表示结点 j 的观察员可以观察到多少人。
分析:
1.对于每一点,考虑它在u到lca和lca到v的路径上两种情况
2.若它再u到lca上应该满足 w[x]+dep[x]=dep[u],否则它应该满足w[x]-dep[x]=dis(u,v)-dep[v] 即w[x]-dep[x]=dep[u]-2*dep[lca]
3.于是问题就变成了,求一棵子树中深度为为一特定值的点数,可以用可持久化线段树,重点是以(2)中等式右半边建树运用树上差分,再区间询问sum
4.注意不要算重

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
using namespace std;
const int maxn=300010;
int to[maxn*2],Next[maxn*2],Begin[maxn],e;
int top[maxn],in[maxn],out[maxn];
int sum[maxn<<2];
int sz,cnt;
int ans[maxn];
int w[maxn];
int n,m;
void read(int& x){
    x=0;
    char c=getchar();
    while(c<'0' || c>'9') c=getchar();
    while(c>='0' && c<='9'){
        x=x*10+c-'0';
        c=getchar();
    }
}
void add(int x,int y){
    to[++e]=y;
    Next[e]=Begin[x];
    Begin[x]=e;
}
struct node{
    int s,t;
    int lca;
}a[maxn];
int dep[maxn],size[maxn],son[maxn],fa[maxn];
void dfs1(int x){
    in[x]=++cnt;
    size[x]=1;
    for(int i=Begin[x];i;i=Next[i]){
        int v=to[i];
        if(v==fa[x]) continue;
        fa[v]=x;
        dep[v]=dep[x]+1;
        dfs1(v);
        size[x]+=size[v];
        if(size[v]>size[son[x]]) son[x]=v;
    }
    out[x]=cnt;
}
void dfs2(int x){
    if(!son[x]) return;
    top[son[x]]=top[x];
    dfs2(son[x]);
    for(int i=Begin[x];i;i=Next[i]){
        int v=to[i];
        if(v==son[x] || v==fa[x]) continue;
        top[v]=v;
        dfs2(v);
    }
}
int LCA(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return x;
}
int t[maxn];
int ch[maxn][2];
int root[maxn];
void init(){
    sz=0;
    memset(root,0,sizeof(root));memset(ch,0,sizeof(ch));memset(sum,0,sizeof(sum));
}
void update(int& o,int l,int r,int pos,int v){
    if(!pos) return;
    if(!o) o=++sz;
    sum[o]+=v;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(pos<=mid) update(ch[o][0],l,mid,pos,v);
    else update(ch[o][1],mid+1,r,pos,v);
}
int query(int o,int l,int r,int ql,int qr){
    if(ql<=l && qr>=r){
        return sum[o];
    }
    int mid=(l+r)>>1;
    int ret=0;
    if(ql<=mid) ret+=query(ch[o][0],l,mid,ql,qr);
    if(qr>mid) ret+=query(ch[o][1],mid+1,r,ql,qr);
    return ret;
}
int main(){
    read(n);read(m);
    int u,v;
    for(int i=1;i<n;i++) read(u),read(v),add(u,v),add(v,u);
    dep[1]=1;top[1]=1;
    dfs1(1);dfs2(1);
    for(int i=1;i<=n;i++) read(w[i]);
    for(int i=1;i<=m;i++) read(a[i].s),read(a[i].t),a[i].lca=LCA(a[i].s,a[i].t);
    for(int i=1;i<=n;i++) t[i]=dep[i]+w[i];
    for(int i=1;i<=m;i++){
        int now=dep[a[i].s];
        update(root[now],1,n,in[a[i].s],1);update(root[now],1,n,in[fa[a[i].lca]],-1);
    }
    for(int i=1;i<=n;i++) ans[i]+=query(root[t[i]],1,n,in[i],out[i]);
    init();
    for(int i=1;i<=n;i++) t[i]=w[i]-dep[i]+n+1;
    for(int i=1;i<=m;i++){
        int now=dep[a[i].s]-2*dep[a[i].lca]+n+1;
        update(root[now],1,n,in[a[i].t],1);update(root[now],1,n,in[a[i].lca],-1);
    }
    for(int i=1;i<=n;i++) ans[i]+=query(root[t[i]],1,n,in[i],out[i]);
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}

^_^

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值