暑假集训日记——8.3( 树链剖分+树形DP)

树链剖分:大佬的博客 大神的博客
题目:【模板】树链剖分
学长的板子:

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
int mod = 1e9+7;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
	//head u的子节点编号,to 子节点, nx 下一子节点的编号
	int head[N], to[N], nx[N], tot, cnt;
	//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
	int dep[N], fa[N], sz[N], son[N], w[N];
	//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
	int top[N], id[N], wt[N];
	//val 维护区间和 ,lz lazy标记
	T val[N<<2], lz[N<<2];
	void init(int n){
		tot = cnt = 0;
		for(int i = 0; i <= n; ++i) head[i] = -1;
	}
	void add(int u, int v){
		to[tot] = v;
		nx[tot] = head[u];
		head[u] = tot++;
	}
	void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
		dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
		son[u] = 0;
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == f) continue;
		    dfs1(v, u);
		    sz[u] += sz[v];
		    if(sz[v] > sz[son[u]]) son[u] = v;
		}
	}
	void dfs2(int u,int topf){//树刨 初始化id wt top
		id[u] = ++cnt;
		wt[cnt] = w[u];
		top[u] = topf;
		if(!son[u]) return;
		dfs2(son[u], topf);
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == fa[u] || v == son[u]) continue;
		    dfs2(v, v);
		}
	}
	void pushdown(int rt,int l,int r){
		if(!lz[rt]) return ;
		int mid = (l+r) >> 1;
		lz[rt<<1] += lz[rt]; lz[rt<<1] %= mod;
		lz[rt<<1|1] += lz[rt]; lz[rt<<1|1] %= mod;
		val[rt<<1] += lz[rt] * (mid-l+1) %mod; val[rt<<1] %= mod;
		val[rt<<1|1] += lz[rt] * (r-mid) %mod; val[rt<<1|1] %=mod;
		lz[rt] = 0;
	}
	void pushup(int rt){
		val[rt] = val[rt<<1] + val[rt<<1|1];
		val[rt] %= mod;
	}
	void build(int rt, int l, int r){
		if(l == r){
		    val[rt] = wt[l];
		    val[rt] %= mod;
		    return ;
		}
		int mid = (l+r) >> 1;
		build(rt<<1, l, mid);
		build(rt<<1|1, mid+1, r);
		pushup(rt);
	}
	T query(int rt, int l, int r, int L, int R){
		if(L <= l && R >= r){
		    return val[rt];
		}
		int mid = (l+r) >> 1; T  ans = 0;
		pushdown(rt, l, r);
		if(L <= mid) ans += query(rt<<1, l, mid, L, R), ans %= mod;
		if(R > mid) ans += query(rt<<1|1, mid+1, r, L, R), ans %= mod;
		pushup(rt);
		return ans;
	}
	T queryRange(int l, int r){//查询l,r的最短路路径和
		T ans = 0;
		while(top[l] != top[r]){
		    if(dep[top[l]] < dep[top[r]]) swap(l, r);
		    ans += query(1, 1, n, id[top[l]], id[l]);
		    ans %= mod;
		    l = fa[top[l]];
		}
		if(dep[l]>dep[r]) swap(l, r);
		ans += query(1, 1, n, id[l], id[r]);
		return ans%mod;
	}
	void update(int rt, int l, int r, int L, int R, int x){
		if(L <= l &&R >= r){
		    lz[rt] += x; lz[rt] %=mod;
		    val[rt] += 1LL*x*(r-l+1)%mod;
		    val[rt] %=mod;
		    return ;
		}
		int mid = (l+r)>>1;
		pushdown(rt, l, r);
		if(L <= mid) update(rt<<1, l, mid, L, R, x);
		if(R > mid) update(rt<<1|1, mid+1, r, L, R, x);
		pushup(rt);
	}
	//修改l,r的最短路径的权值
	void updateRange(int l, int r, int x){
		while(top[l] != top[r]){
		    if(dep[top[l]] < dep[top[r]]) swap(l, r);
		    update(1, 1, n, id[top[l]], id[l], x);
		    l = fa[top[l]];
		}
		if(dep[l]>dep[r]) swap(l, r);
		update(1, 1, n, id[l], id[r], x);
	}
	//修改子树权值
	void updateSon(int x,int y){
		update(1, 1, n, id[x], id[x]+sz[x]-1, y);
	}
	//查询子树和
	T querySon(int x){
		return query(1, 1, n, id[x], id[x]+sz[x]-1);
	}
};
Heavy_Light_Decomposition<long long>hld;//模板
int main(){
	//n个点,m次操作,r是根节点,p模
	scanf("%d%d%d%d", &n, &m, &r, &p);
	hld.init(n);
	mod = p;
	for(int i = 1; i <= n; ++i){
		scanf("%d", &hld.w[i]);
	}
	for(int i = 1; i < n; ++i){
		int u, v; scanf("%d%d", &u, &v);
		hld.add(u, v);
		hld.add(v, u);
	}
	hld.dfs1(r, 0);
	hld.dfs2(r, r);
	hld.build(1, 1, n);
	for(int i = 1; i <= m; ++i){
		int id, x, y, z;
		scanf("%d", &id);
		if(id == 1){
			scanf("%d%d%d", &x, &y, &z);
			hld.updateRange(x, y, z);
		}
		else if(id == 2){
			scanf("%d%d", &x, &y);
			printf("%lld\n", hld.queryRange(x, y));
		}
		else if(id == 3){
			scanf("%d%d", &x, &y);
			hld.updateSon(x, y);
		}
		else scanf("%d", &x), printf("%lld\n", hld.querySon(x));
	}
	return 0;
}

A - Aragorn’s Story
题解:板子题(awsl,lazy标记忘记初始化了…所以总是不过
用线段树做总是不对…用树状数组做了一遍

#pragma comment(linker, "/STACK:10240000,10240000")
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e6+10;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
	//head u的子节点编号,to 子节点, nx 下一子节点的编号
	int head[N], to[N], nx[N], tot, cnt;
	//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
	int dep[N], fa[N], sz[N], son[N], w[N];
	//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
	int top[N], id[N], wt[N],tree[N];
	//val 维护区间和 ,lz lazy标记
	T val[N<<2], lz[N<<2];
	void init(int n){
		tot = cnt = 0;
		for(int i = 0; i <= n; ++i) head[i] = -1;
	}
	void add(int u, int v){
		to[tot] = v;
		nx[tot] = head[u];
		head[u] = tot++;
	}
	void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
		dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
		son[u] = 0;
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == f) continue;
		    dfs1(v, u);
		    sz[u] += sz[v];
		    if(sz[v] > sz[son[u]]) son[u] = v;
		}
	}
	void dfs2(int u,int topf){//树刨 初始化id wt top
		id[u] = ++cnt;
		wt[cnt] = w[u];
		top[u] = topf;
		if(!son[u]) return;
		dfs2(son[u], topf);
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == fa[u] || v == son[u]) continue;
		    dfs2(v, v);
		}
	}
	int lowbit(int x)
    {
        return x&(-x);
    }

    void ADD(int p,int v)
    {
        for(int i=p;i<=n;i+=lowbit(i))
            tree[i]+=v;
    }

    int SUM(int p)
    {
        int ret=0;
        for(int i=p;i;i-=lowbit(i))
            ret+=tree[i];
        return ret;
    }


    void Change(int u,int v,int K)
    {
        int f1=top[u],f2=top[v];
        while(f1!=f2)
        {
            if(dep[f1]<dep[f2])
            {
                swap(f1,f2);
                swap(u,v);
            }
            ADD(id[f1],K);
            ADD(id[u]+1,-K);
            u=fa[f1];
            f1=top[u];
        }
        if(dep[u]<dep[v]) swap(u,v);
        ADD(id[v],K);
        ADD(id[u]+1,-K);
    }

};
Heavy_Light_Decomposition<int>hld;//模板
int main(){

	while(scanf("%d%d%d", &n, &p, &m)!=EOF)
    {
        r=1;
        hld.init(n);
        for(int i = 1; i <= n; ++i){
            scanf("%d", &hld.w[i]);
        }
        for(int i = 1; i <= p; ++i){
            int u, v; scanf("%d%d", &u, &v);
            hld.add(u, v);
            hld.add(v, u);
        }
        hld.dfs1(r, 0);
        hld.dfs2(r, r);
        memset(hld.tree,0,sizeof(hld.tree));

        for(int i=1;i<=n;i++)
        {
            hld.ADD(hld.id[i],hld.w[i]);
            hld.ADD(hld.id[i]+1,-hld.w[i]);
        }
        for(int i = 1; i <= m; ++i){
            int id, x, y, z;
            char str[5];
            //getchar();
            scanf("%s", str);
            if(str[0] == 'I'){
                scanf("%d%d%d", &x, &y, &z);
                hld.Change(x, y, z);
            }
            else if(str[0] == 'Q'){
                scanf("%d", &x);
                printf("%d\n", hld.SUM(hld.id[x]));
            }
            else if(str[0] == 'D'){
                scanf("%d%d%d", &x, &y, &z);
                hld.Change(x, y, -z);
            }
        }
    }

}


Housewife Wind
题解:树刨,边化点
模板开太大…没有意识到,然后就疯狂的TTTTT,以及不要用 v e c t o r vector vector,慢的要死

#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 2e5+10;
int mod = 1e9+7;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
	//head u的子节点编号,to 子节点, nx 下一子节点的编号
	int head[N], to[N], nx[N], tot, cnt;
	//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
	int dep[N], fa[N], sz[N], son[N], w[N];
	//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
	int top[N], id[N], wt[N];
	//val 维护区间和 ,lz lazy标记
	T val[N<<2], lz[N<<2];
	void init(int n){
		tot = cnt = 0;
		for(int i = 0; i <= n; ++i) head[i] = -1;
	}
	void add(int u, int v){
		to[tot] = v;
		nx[tot] = head[u];
		head[u] = tot++;
	}
	void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
		dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == f) continue;
		    dfs1(v, u);
		    sz[u] += sz[v];
		    if(sz[v] > sz[son[u]]) son[u] = v;
		}
	}
	void dfs2(int u,int topf){//树刨 初始化id wt top
		id[u] = ++cnt;
		wt[cnt] = w[u];
		top[u] = topf;
		if(!son[u]) return;
		dfs2(son[u], topf);
		for(int i = head[u]; ~i; i = nx[i]){
		    int v = to[i];
		    if(v == fa[u] || v == son[u]) continue;
		    dfs2(v, v);
		}
	}
	void pushup(int rt){
		val[rt] = val[rt<<1] + val[rt<<1|1];
		val[rt] %= mod;
	}
	void build(int rt, int l, int r){
		if(l == r){
		    val[rt] = wt[l];
		    val[rt] %= mod;
		    return ;
		}
		int mid = (l+r) >> 1;
		build(rt<<1, l, mid);
		build(rt<<1|1, mid+1, r);
		pushup(rt);
	}
	T query(int rt, int l, int r, int L, int R){
		if(L <= l && R >= r){
		    return val[rt];
		}
		int mid = (l+r) >> 1; T  ans = 0;
		//pushdown(rt, l, r);
		if(L <= mid) ans += query(rt<<1, l, mid, L, R), ans %= mod;
		if(R > mid) ans += query(rt<<1|1, mid+1, r, L, R), ans %= mod;
		pushup(rt);
		return ans;
	}
	T queryRange(int l, int r){//查询l,r的最短路路径和
		T ans = 0;
		while(top[l] != top[r]){
		    if(dep[top[l]] < dep[top[r]]) swap(l, r);
		    ans += query(1, 1, n, id[top[l]], id[l]);
		    ans %= mod;
		    l = fa[top[l]];
		}
		/********边化为点***********/
		if(l == r) return ans;///特判不移动
		if(dep[l]>dep[r]) swap(l, r);
		///由于边变成点,改为id[son[l]]
		ans += query(1, 1, n, id[son[l]], id[r]);
		return ans%mod;
	}
	void update(int rt, int l, int r, int L, int R, int x){
		if(L <= l &&R >= r){
		    //lz[rt] += x; lz[rt] %=mod;
		    val[rt] = 1LL*x%mod;
		    val[rt] %=mod;
		    return ;
		}
		int mid = (l+r)>>1;
		//pushdown(rt, l, r);
		if(L <= mid) update(rt<<1, l, mid, L, R, x);
		if(R > mid) update(rt<<1|1, mid+1, r, L, R, x);
		pushup(rt);
	}
	//修改l,r的最短路径的权值
	void updateRange(int l, int r, int x){
		while(top[l] != top[r]){
		    if(dep[top[l]] < dep[top[r]]) swap(l, r);
		    update(1, 1, n, id[top[l]], id[l], x);
		    l = fa[top[l]];
		}
		if(dep[l]>dep[r]) swap(l, r);
		update(1, 1, n, id[l], id[r], x);
	}
};

struct note
{
    int u,v,w;
}h[N];

Heavy_Light_Decomposition<int>hld;//模板
int main(){
	scanf("%d%d%d", &n, &m, &p);
	r=1;
	hld.init(n);
	for(int i = 1; i < n; ++i){
		int u, v,w; scanf("%d%d%d", &u, &v,&w);
		h[i].u=u,h[i].v=v,h[i].w=w;
		hld.add(u, v);
		hld.add(v, u);
	}
	hld.dfs1(r, 0);
	hld.dfs2(r, r);
	hld.build(1, 1, n);
	int xx;
	for(int i=1;i<n;i++){///比较一条边上哪个点的深度大,就把权值加在哪个点上
         if(hld.dep[h[i].u]<hld.dep[h[i].v]) xx=h[i].v;
         else xx=h[i].u;
         hld.updateRange(xx,xx,h[i].w);
     }
	for(int i = 1; i <= m; ++i){
		int id, x, y, z;
		//getchar();
		scanf("%d", &id);
		if(id == 1){
			scanf("%d%d", &x, &y);
			if(hld.dep[h[x].u]<hld.dep[h[x].v]) xx=h[x].v;
            else xx=h[x].u;
            hld.updateRange(xx,xx,y);
		}
		else if(id == 0){
			scanf("%d", &x);
			printf("%d\n", hld.queryRange(x, p));
			p=x;
		}
	}
	return 0;
}

B - Valid Sets
题意:
给定一棵树,任意选定节点,节点间是联通的(也可以是同一个节点),使这些个点的路径上的点中最大值和最小值小于 d d d
题解:树形 d p dp dp
枚举节点作为最大节点,然后遍历它所控制的范围计算符合条件的方案数

#include<bits/stdc++.h>
#define mp make_pair
#define maxn 10000000
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef long double ld;

const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const ll mod=998244353 ;
int n,m,x,y,k,cnt;

int d,a[maxn],root;
const int MOD=1e9+7;
vector<int>G[maxn];
long long dfs(int u,int fa)
{
	long long res=1;
	int v;
	for(int i=0;i<(int)G[u].size();i++)
	{
		v=G[u][i];
		if(v==fa)///单向
			continue;
		if(a[v]>a[root])///使得root最大,不需要考虑最小值
			continue;
		if(a[v]==a[root]&&v>root)///避免重复
			continue;
		if(a[root]-a[v]>d)///最小的节点都满足的话其他值都满足
			continue;
		res=res*(dfs(v,u)+1)%MOD;///计算子树的数目
	}
	return res;
}
int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	scanf("%d%d",&d,&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	long long ans=0;
	///枚举每个结点,把每个节点都当作最大节点来进行计算
	for(int i=1;i<=n;i++)
	{
		root=i;
		ans=(ans+dfs(root,-1))%MOD;
	}
	printf("%I64d\n",ans);
	//system("pause");
	return 0;
}

D. Choosing Capital for Treeland
题意:
这道题给出n个节点的树,其中的边是有向边,让你从其中选择一个节点,使得从该节点出发能够到达其他所有节点,中途可以改变边的方向,但需要使改变的次数最少。
题解:树形 d p dp dp
先预处理一个点 u u u,将 u u u 点作为根进行遍历,记录反向的个数 w w w
会发现一个很有趣的规律;
当将其他结点v 作为根时的反向边的个数为 n n n
n = w − 2 ∗ ( u 到 v 路 径 上 的 反 向 边 ) + ( u 到 v 路 径 长 度 ) n=w -2*(u到v路径上的反向边)+ (u到v路径长度) n=w2(uv)+(uv)
然后找出所有节点的最小值就可以了

#include<bits/stdc++.h>
#define mp make_pair
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef long double ld;

const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const ll mod=998244353 ;
int n,m,x,y,k,cnt;

struct note
{
    int u,v,w;
}h;

vector<note>E[N];
int dp[N],dep[N],ans[N],pop=INF;

void dfs1(int x,int fa)
{
    dep[x]=dep[fa]+1;
    for(int i=0;i<E[x].size();i++)
    {
        note y;
        y=E[x][i];
        if(y.v==fa) continue;
        if(y.w<0) cnt++;
        dp[y.v]=dp[y.u]+((y.w==-1)?1:0);///被三元符坑了忘记加括号...
        dfs1(y.v,y.u);
    }
}

int main()
{
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    scanf("%d", &n);
    for(int i=0;i<n-1;i++)
    {
          int u,v;
          scanf("%d%d",&u,&v);
          h.u=u,h.v=v,h.w=1;
          E[u].push_back(h);
          h.u=v,h.v=u,h.w=-1;
          E[v].push_back(h);
    }
    dep[0]=-1;
    dfs1(1,0);
    ans[1]=cnt;
    pop=min(pop,ans[1]);
    for(int i=2;i<=n;i++)
    {
        ans[i]=ans[1]-2*dp[i]+dep[i];
        pop=min(pop,ans[i]);
        //printf("%d\n",dp[i]);
    }
    printf("%d\n",pop);
    for(int i=1;i<=n;i++)
    {
        if(pop==ans[i])
            printf("%d ",i);
    }




}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值