• [JLOI2014]松鼠的新家 T22 D71


    [JLOI2014]松鼠的新家 T22 D71

    [ JLOI2014]松鼠的新家

    思路

    树剖。

    遍历数组,将数组中相邻节点的路径权值+1。

    输出答案时线段树单点询问

    #include<bits/stdc++.h>
    #define ll long long
    #define pii pair<int,int>
    #define fi first
    #define se second
    #define pb push_back
    #define si size()
    #define ls (p<<1)
    #define rs ((p<<1)|1)
    #define mid (t[p].l+t[p].r)/2 
    using namespace std;
    ll read(){ll x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
    inline void Prin(ll x){if(x < 0){putchar('-');x = -x;}if(x > 9) Prin(x / 10);putchar(x % 10 + '0');}
    const int qs=3e5+7;
    
    const int mod=1e9;
    ll n,m,q,a[qs]; 
    int dep[qs],f[qs],fr[qs],sz[qs],son[qs],top[qs],cnt=0;
    vector<int> v[qs];
    pii id[qs];
    struct Tree{
    	ll val,add;
    	int l,r;
    	#define l(x) t[x].l
    	#define r(x) t[x].r
    	#define val(x) t[x].val
    	#define add(x) t[x].add
    }t[qs<<2];
    
    void pushup(int p){ val(p)=(val(ls)+val(rs))%mod;}
    void down(int p){
    	if(!add(p)) return;
    	val(ls)=(val(ls)+add(p)*(r(ls)-l(ls)+1))%mod;
    	val(rs)=(val(rs)+add(p)*(r(rs)-l(rs)+1))%mod;
    	add(ls)=(add(p)+add(ls))%mod;
    	add(rs)=(add(p)+add(rs))%mod;
    	add(p)=0;
    }
    void build(int p,int l,int r){
    	l(p)=l,r(p)=r;add(p)=0;
    	if(l==r){
    		val(p)=0;
    		return;
    	}
    	build(ls,l,mid);
    	build(rs,mid+1,r);
    	pushup(p);
    //	cout<<"l="<<l<<" r="<<r<<" val="<<val(p)<<"
    ";
    }
    
    void update(int p,int l,int r,ll val){
    	if(l<=l(p)&&r>=r(p)){
    		val(p)=(val(p)+val*(r(p)-l(p)+1))%mod;
    		add(p)=(val+add(p))%mod;
    		//cout<<"l="<<l(p)<<" r="<<r(p)<<" val="<<val(p)<<"
    ";
    		return;
    	}
    	down(p);
    	if(l<=mid) update(ls,l,r,val);
    	if(r>mid) update(rs,l,r,val);
    	pushup(p);
    }
    
    ll ask(int p,int l,int r){
    	if(l<=l(p)&&r>=r(p)) return val(p);
    	down(p);
    	ll val=0;
    	if(l<=mid) val=(val+ask(ls,l,r))%mod;
    	if(r>mid) val=(val+ask(rs,l,r))%mod;
    	return val; 
    }
    
    void dfs(int x,int fa){
    	dep[x]=dep[fa]+1; f[x]=fa; sz[x]=1; son[x]=0;
    	int ms=0;
    	for(int i=0;i<v[x].si;++i){
    		int p=v[x][i];
    		if(p==fa) continue;
    		dfs(p,x);
    		sz[x]+=sz[p];
    		if(sz[p]>ms) son[x]=p,ms=sz[p];
    	}
    }
    
    void dfn(int x,int po){
    	id[x].fi=++cnt;
    	fr[cnt]=x;
    	top[x]=po;
    	if(!son[x]) {
    		id[x].se=cnt;	return;
    	}
    	dfn(son[x],po);
    	for(int i=0;i<v[x].si;++i){
    		int p=v[x][i];
    		if(p==f[x]||p==son[x]) continue;
    		dfn(p,p);
    	}
    	id[x].se=cnt;
    }
    
    
    void updRange(int x,int y,ll k){
    	k%=mod;
    	while(top[x]!=top[y]){
    		if(dep[top[x]]<dep[top[y]]) swap(x,y);
    		update(1,id[top[x]].fi,id[x].fi,k);
    		x=f[top[x]];
    	}
    	if(dep[x]>dep[y]) swap(x,y);
    	update(1,id[x].fi,id[y].fi,k);
    }
    
    ll qRange(int x,int y){
    	ll ans=0,res;
    	while(top[x]!=top[y]){
    		if(dep[top[x]]<dep[top[y]]) swap(x,y);
    		res=ask(1,id[top[x]].fi,id[x].fi);
    		ans=(ans+res)%mod;
    		x=f[top[x]];
    	}
    	if(dep[x]>dep[y]) swap(x,y);
    	res=ask(1,id[x].fi,id[y].fi);
    	ans=(ans+res)%mod;
    	return ans;
    }
    
    int main(){
    	
    	scanf("%lld
    ",&n);
    	ll x,y;
    	for(int i=1;i<=n;++i){
    		scanf("%lld",&a[i]);
    	}
    	for(int i=1;i<n;++i){
    		scanf("%lld%lld",&x,&y);
    		v[x].pb(y);
    		v[y].pb(x);
    	}
    	dfs(1,0);
    	dfn(1,1);
    	build(1,1,n);
    	for(int i=2;i<=n;++i){
    		updRange(a[i-1],a[i],1);
    	}
    	for(int i=1;i<=n;++i){
    		int ans=ask(1,id[i].fi,id[i].fi);
    		if(i!=a[1]) ans--;
    		cout<<ans<<"
    ";
    	}
    
    	return 0;
    }
    
    
    /*
    
    */
    
    
    
    
    
    
  • 相关阅读:
    wpf Behavior
    wpf Trigger
    语法糖
    Lambda 表达式
    wpf 3D动画
    IEnumerable接口学习
    Delegates, Events, and Anonymous Methods 委托、事件与匿名方法
    wpf 平滑效果随记
    软件工程第一篇博客
    记考研高数第一课
  • 原文地址:https://www.cnblogs.com/Suki-Sugar/p/15435137.html
Copyright © 2020-2023  润新知