CF1009 长链剖分优化树形DP或线段树合并

最新推荐文章于 2022-09-20 17:32:02 发布
最新推荐文章于 2022-09-20 17:32:02 发布
阅读量213 收藏
点赞数
分类专栏： 数据结构 DP
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Zenith_Habitant/article/details/108303128
版权 
//#define LOCAL
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mem(a, b) memset(a,b,sizeof(a))
#define sz(a) (int)a.size()
#define INF 0x3f3f3f3f
#define DNF 0x7f7f7f7f
#define DBG printf("this is a input\n")
#define fi first
#define se second
#define PiL pair <ll , ll>
#define PLL pair <ll , ll>
#define Pii pair <int , int>
#define mk(a, b) make_pair(a,b)
#define pb push_back
#define LF putchar('\n')
#define SP putchar(' ')
#define p_queue priority_queue
#define CLOSE ios::sync_with_stdio(0); cin.tie(0)

template<typename T>
void read(T &x) {x = 0;char ch = getchar();ll f = 1;while(!isdigit(ch)){if(ch == '-')f *= -1;ch = getchar();}while(isdigit(ch)){x = x * 10 + ch - 48; ch = getchar();}x *= f;}
template<typename T, typename... Args>
void read(T &first, Args& ... args) {read(first);read(args...);}
template<typename T>
void write(T arg) {T x = arg;if(x < 0) {putchar('-'); x =- x;}if(x > 9) {write(x / 10);}putchar(x % 10 + '0');}
template<typename T, typename ... Ts>
void write(T arg, Ts ... args) {write(arg);if(sizeof...(args) != 0) {putchar(' ');write(args ...);}}

using namespace std;

const int N = 1e6+5;
int head[N] , cnt, n;
struct node
{
    int next , t , w;
}edge[2 * N];
void add (int f, int t, int w)
{
    edge[cnt].w = w;
    edge[cnt].t = t;
    edge[cnt].next = head[f];
    head[f] = cnt ++;
}
int h[N], son[N];
int *dp[N], temp[N], *cur = temp;
int ans[N];
void dfs1 (int u , int fa)
{
    h[u] = 1;
    for (int i = head[u] ; i != -1 ; i = edge[i].next)
    {
        int v = edge[i].t;
        if (v != fa)
        {
            dfs1 (v , u);
            h[u] = max (h[u], h[v] + 1);
            if (h[v] > h[son[u]]) son[u] = v;
        }
    }
}
void dfs2 (int u , int fa)
{
    dp[u][0] = 1;
    if (son[u])
    {
        dp[son[u]] = dp[u] + 1; //共享内存
        dfs2(son[u], u);
        ans[u] = ans[son[u]] + 1;
    }
    for (int i = head[u] ; i != -1 ; i = edge[i].next)
    {
        int v = edge[i].t;
        if (v != son[u] && v != fa)
        {
            dp[v] = cur, cur += h[v]; // 分配内存
            dfs2(v, u);
            for (int j = 1 ; j <= h[v] ; j ++)
            {
                dp[u][j] += dp[v][j-1];
                if (dp[u][j] > dp[u][ans[u]] || (dp[u][j] == dp[u][ans[u]] && j < ans[u])) ans[u] = j;
            }
        }
    }
    if (dp[u][ans[u]] == 1) ans[u] = 0;
}
int main()
{
    read (n);
    mem(head, -1);
    for (int i = 1 ; i < n ; i ++)
    {
        int u , v;
        read (u , v);
        add (u, v, 1);
        add (v, u, 1);
    }
    dfs1 (1 , 1);
    dp[1] = cur , cur += h[1]; //分配内存
    dfs2 (1 , 1);
    for (int i = 1 ; i <= n ; i ++)
        write (ans[i]), LF;
}

以上为长链剖分优化树形DP

//#define LOCAL
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mem(a, b) memset(a,b,sizeof(a))
#define sz(a) (int)a.size()
#define INF 0x3f3f3f3f
#define DNF 0x7f7f7f7f
#define DBG printf("this is a input\n")
#define fi first
#define se second
#define PiL pair <ll , ll>
#define PLL pair <ll , ll>
#define Pii pair <int , int>
#define mk(a, b) make_pair(a,b)
#define pb push_back
#define LF putchar('\n')
#define SP putchar(' ')
#define p_queue priority_queue
#define CLOSE ios::sync_with_stdio(0); cin.tie(0)

template<typename T>
void read(T &x) {x = 0;char ch = getchar();ll f = 1;while(!isdigit(ch)){if(ch == '-')f *= -1;ch = getchar();}while(isdigit(ch)){x = x * 10 + ch - 48; ch = getchar();}x *= f;}
template<typename T, typename... Args>
void read(T &first, Args& ... args) {read(first);read(args...);}
template<typename T>
void write(T arg) {T x = arg;if(x < 0) {putchar('-'); x =- x;}if(x > 9) {write(x / 10);}putchar(x % 10 + '0');}
template<typename T, typename ... Ts>
void write(T arg, Ts ... args) {write(arg);if(sizeof...(args) != 0) {putchar(' ');write(args ...);}}

using namespace std;

const int N = 1e6+5;
int n;
int root[N] , lc[25*N], rc[25*N], val[25*N], id[25*N], tot;
int head[N], cnt;
int ans[N];
struct node {
    int t, next , w;
}edge[N * 2];
void add (int f, int t, int w)
{
    edge[cnt].t = t;
    edge[cnt].w = w;
    edge[cnt].next = head[f];
    head[f] = cnt ++;
}
int dep[N] = {1, 1};
void dfs1 (int u , int fa)
{
    for (int i = head[u] ; i != -1 ; i = edge[i].next)
    {
        int v = edge[i].t;
        if (v != fa)
        {
            dep[v] = dep[u] + 1;
            dfs1 (v, u);
        }
    }
}
void push_up(int i)
{
    if (val[lc[i]] >= val[rc[i]])
    {
        val[i] = val[lc[i]];
        id[i] = id[lc[i]];
    }
    else
    {
        val[i] = val[rc[i]];
        id[i] = id[rc[i]];
    }
}
void insert (int &rt, int l , int r, int x , int add)
{
    if (!rt) rt = ++ tot;
    if (l == r)
    {
        id[rt] = x;
        val[rt] += add;
        return ;
    }
    int mid = (l + r) >> 1;
    if (x <= mid)
        insert (lc[rt], l , mid , x , add);
    else
        insert (rc[rt], mid + 1, r , x , add);
    push_up(rt);
}
int merge (int x, int y, int l , int r)
{
    if (!x) return y;
    if (!y) return x;
    if (l == r)
    {
        val[x] += val[y];
        id[x] = l;
        return x;
    }
    int mid = (l + r) >> 1;
    lc[x] = merge(lc[x], lc[y], l, mid);
    rc[x] = merge(rc[x], rc[y], mid+1, r);
    push_up(x);
    return x;
}
void dfs (int u , int fa)
{
    insert (root[u], 1, n, dep[u], 1);
    for (int i = head[u] ; i != -1 ; i = edge[i].next)
    {
        int v = edge[i].t;
        if (v != fa)
        {
            dfs (v, u);
            root[u] = merge(root[u], root[v], 1, n);
        }
    }
    ans[u] = id[root[u]] - dep[u];
}
int main()
{
    read (n);
    mem(head,-1);
    for (int i = 1 ; i < n; i ++)
    {
        int u , v;
        read (u , v);
        add (u , v , 1);
        add (v , u , 1);
    }
    dfs1 (1 , 1);
    dfs (1 , 1);
    for (int i = 1 ; i <= n ; i ++)
        write (ans[i]), LF;
}

以上为动态开点权值线段树合并

//#define LOCAL
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mem(a, b) memset(a,b,sizeof(a))
#define sz(a) (int)a.size()
#define INF 0x3f3f3f3f
#define DNF 0x7f
#define DBG printf("this is a input\n")
#define fi first
#define se second
#define mk(a, b) make_pair(a,b)
#define pb push_back
#define LF putchar('\n')
#define SP putchar(' ')
#define p_queue priority_queue
#define CLOSE ios::sync_with_stdio(0); cin.tie(0)

template<typename T>
void read(T &x) {x = 0;char ch = getchar();ll f = 1;while(!isdigit(ch)){if(ch == '-')f *= -1;ch = getchar();}while(isdigit(ch)){x = x * 10 + ch - 48; ch = getchar();}x *= f;}
template<typename T, typename... Args>
void read(T &first, Args& ... args) {read(first);read(args...);}
template<typename T>
void write(T arg) {T x = arg;if(x < 0) {putchar('-'); x =- x;}if(x > 9) {write(x / 10);}putchar(x % 10 + '0');}
template<typename T, typename ... Ts>
void write(T arg, Ts ... args) {write(arg);if(sizeof...(args) != 0) {putchar(' ');write(args ...);}}
using namespace std;

ll gcd(ll a, ll b) {
    return b == 0 ? a : gcd(b, a % b);
}

ll lcm(ll a, ll b) {
    return a / gcd(a, b) * b;
}
const int N = 1e6 + 5;
int n , m;
vector <int> edge[N];
vector <pair<int,int> > q[N];
int ans[N], siz[N] , son[N] , dep[N];
int cnt[N], vis[N], maxCnt , maxId;
void dfs1 (int u , int fa)
{
    siz[u] = 1;
    for (auto v : edge[u])
    {
        if (v != fa)
        {
            dep[v] = dep[u] + 1;
            dfs1 (v, u);
            siz[u] += siz[v];
            if (siz[son[u]] < siz[v])
                son[u] = v;
        }
    }
}
void calc (int u , int fa, int k)
{
    cnt[dep[u]] += k;
    if (cnt[dep[u]] > maxCnt)
        maxId = dep[u], maxCnt = cnt[dep[u]];
    else if (cnt[dep[u]] == maxCnt)
        maxId = min (maxId, dep[u]);
    for (auto v : edge[u])
        if (v != fa && !vis[v])
            calc(v , u, k);
}
void dfs2 (int u , int fa , int opt)
{
    for (auto v : edge[u])
        if (v != fa && v != son[u])
            dfs2 (v , u , 0);
    if (son[u]) dfs2 (son[u] , u , 1), vis[son[u]] = 1;
    calc (u , fa , 1);
    ans[u] = maxId - dep[u];
    if (son[u]) vis[son[u]] = 0;
    if (!opt) calc(u, fa, -1), maxCnt = maxId = 0;
}
int main()
{
    read (n);
    for (int i = 1 ; i < n ; i ++)
    {
        int u , v;
        read (u , v);
        edge[u].pb (v);
        edge[v].pb (u);
    }
    dep[1] = 1;
    dfs1 (1, 1);
    dfs2 (1, 0, 1);
    for (int i = 1 ; i <= n ; i ++)
        write(ans[i]), LF;
}

以上是Dsu On Tree

卡常带师
关注
专栏目录
cf1009F Dominant Indices 长链剖分
BPM136(BPMThor)
08-17 209
终于入坑长链剖分了。以前只知道思想但是不会写代码。（因为有些细节没想明白，突然今早睡醒在梦中理解了x 因为要O(1)O(1)O(1)继承重儿子的信息，我们可以使用主席树/指针/vector来实现。 主席树的话就是直接把重儿子的链直接给就完事了。 指针的话非常轻松，每次直接给这条链分配一块内存，每次重儿子移位就好了，轻儿子的内存重新分配。 vector的话可能需要倒序，有点蛋疼，每次直接使用重儿子的...
树形dp（二叉苹果树，选课，最长链，战略游戏）
avenLJY的博客
05-18 314
树形dp 【例1】二叉苹果树 题意：有一棵苹果树，如果树枝有分叉，一定是分2叉（就是说没有只有1个儿子的结点）。这棵树共有N个结点（叶子点或者树枝分叉点），编号为1–N,树根编号一定是1。我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。给定需要保留的树枝数量，求出最多能留住多少苹果。 step1:把边权转为点权，记在儿子处 step...
2018.11.03 NOIP模拟 树（长链剖分优化dp）
weixin_30642561的博客
11-03 111
传送门 考虑直接推式子不用优化怎么做。 显然每一个二进制位分开计算贡献就行。 即记录fi,jf_{i,j}fi,j​表示距离iii这个点不超过jjj的点的每个二进制位的0/10/10/1个数。 但直接存是会爆炸的。 考虑到每个数只会被用一次，所以可以考虑主席树那种复用信息的思想来继承长链后代的信息，然后短链直接暴力统计贡献就行。 由于ldxldxld...
长链剖分优化树形dp
caoyang1123的博客
05-19 698
apio铁牌告辞（开场想打暴力然后gedit码代码5个小时没写完三题最低档暴力真是快乐），听课也就学到了一丢丢这个东西。 模板题： https://www.luogu.org/problemnew/show/P5384 首先k级兄弟可以一遍dfs的时候丢到k级父亲上变成求k级孩子的询问。 求k级孩子有个很简单的做法，直接dfs的时候维护扫到某个点时记录下的每种dep出现次数，对于所有求这个...
长链剖分优化dp三例题
daxian3723的博客
03-23 435
　　首先，重链剖分我们有所认识，在dsu on tree和数据结构维护链时我们都用过他的性质。 　　在这里，我们要介绍一种新的剖分方式，我们求出这个点到子树中的最长链长，这个链长最终从哪个儿子更新而来，那个儿子就是所谓的“重儿子”，也可以叫长儿子。 　　我们的做法就是，在统计一个点的信息时，对于重儿子，我们直O(1)接继承它的答案（这里有指针技巧，只能看代码，不可言传），对于轻...
bzoj3653: 谈笑风生（长链剖分优化dp）
苟为蒟蒻又何妨
01-19 474
传送门 长链剖分优化dpdpdp水题。 题意简述：给一棵树，mmm次询问，每次给一个点aaa和一个值kkk，询问满足如下条件的三元组(a,b,c)(a,b,c)(a,b,c)的个数。 a,b是c的祖先 a,b的距离不超过k 思路： 考虑单独处理每一个询问怎么做。 显然a,ba,ba,b的位置关系有两种。 bbb是aaa的祖先，此时ccc一定在aaa子树中，这种情况的三元组个数是(sizea...
线段树（合并/分治）&&重/长链剖分&&主席树专题（持续更新中）
苟为蒟蒻又何妨
07-28 3202
常规线段树（简单） 模板 树剖区间取负，单点修改维护最大最小总和 单点修改维护根节点乘积 维护区间连续(0/1)(0/1)(0/1)的长度 根据逆序对数还原排列 双指针+线段树永久化标记 505050多棵线段树动态开点 线段树（树状数组）求逆序对 单点插入区间查询 区间开根区间求和 单点修改区间最大连续和 线段树与算法（简单） 优化dpdpdp 线段树合并 李超线...
BZOJ4543 POI2014 Hotel加强版 【长链剖分】【DP】*
Dream_maker的博客
08-10 387
BZOJ4543 POI2014 Hotel加强版 Description 同OJ3522 数据范围：n&lt;=100000 Sample Input 7 1 2 5 7 2 5 2 3 5 6 4 5 Sample Output 5 我处理DP问题就像是智障一样 感谢yyf大神的讲解啊 首先我们可以发现三个点两两距离相等说明三个点的连线交点是...
2020牛客暑期多校训练营Tokens on the Tree(树形DP，树链剖分)
ZCETHAN 的博客
08-06 290
Tokens on the Tree 题目描述 输入描述: 输出描述: 示例1 输入 2 5 1 2 3 3 10 1 2 3 4 3 6 3 8 2 输出 71 989 说明 题目大意 给定一棵树，这棵树上有黑白两种颜色，白色和黑色节点各构成了一坨连通块，其中一个节点可以将其颜色转移到它所在的连通块的外面一圈。 再定义一个函数f(u,v)f(u,v)f(u,v)，如果一个连通块通过变换能使得这个块的两端互换位置，那么称这两个形态为同种形态。现要求对于所有的u+v≤nu+v\le nu+v≤n，nn
【luogu CF1009F】Dominant Indices（长链剖分优化DP）
欢迎您的光顾awa
04-01 247
给你一棵以 1 为根的树，设 d(u,x) 为 u 子树总到 u 距离为 x 的点数。 然后要你对于每个点 x 都要求一个最小的 k 使得 d(x,k) 最大。
2019.01.19 bzoj3653: 谈笑风生（长链剖分优化dp）
weixin_30535167的博客
01-19 156
传送门 长链剖分优化dpdpdp水题。 题意简述：给一棵树，mmm次询问，每次给一个点aaa和一个值kkk，询问满足如下条件的三元组(a,b,c)(a,b,c)(a,b,c)的个数。 a,b是c的祖先 a,b的距离不超过k 思路： 考虑单独处理每一个询问怎么做。 显然a,ba,ba,b的位置关系有两种。 bbb是aaa的祖先，此时ccc一定在...
长链剖分优化树上dp
rgndao的博客
04-01 434
长链剖分可以把维护子树中只与深度有关的信息做到线性的时间复杂度。 例题cf1009f 给一棵树，定义每个点的dom值为，以该点为根的子树重中，节点数最多的那一层的层数，即那一层距离这个根节点有几条边，如果多层节点数相同，取最小的层数。例如单独叶节点的dom值是0，一条垂直的链的dom值也是0（每层数量都是1，取最前面的） 定义dp[i...
bzoj4543 长链剖分优化树形DP
weixin_45539557的博客
03-15 362
bzoj4543 长链剖分优化树形DP 题意： 树上统计三元组(i,j,k)两两距离相等的数目 思路： 看起来有两种，但合并后其实只有一种 前面两种其实只是j=0和j=dj=0和j=dj=0和j=d的特例而已 想清楚这点非常重要，这决定我们能否顺利推出方程，所以我们需要的就是两个东西 f[i][j]f[i][j]f[i][j]表示以i为根距离i距离为j的有多少个，容易想到转移f[u][j]+=f[v][j−1]f[u][j]+=f[v][j-1]f[u][j]+=f[v][j−1]， g[i][j]g[
中心城镇问题（长链剖分优化树形dp）
01-20 295
动态数组毁我青春
CF-1009F Dominant Indices：长链剖分+DP指针优化
SHUlpt的博客
02-04 324
Dominant Indices【长链剖分+DP指针优化】 给定一棵以 1 为根，共有 nnn 个节点的树。设 d(u,x)d(u,x)d(u,x) 为 uuu 子树中到 uuu 距离为 xxx 的节点数。 对于每个点，求一个最小的 kkk，使得 d(u,x)d(u,x)d(u,x) 最大 题解 令 fu,depf_{u,dep}fu,dep​ 表示 uuu 的子树中与 uuu 的距离为 depdepdep 的点的个数，则有转移方程 fu,dep=∑v∈sonufv,dep−1 f_{u,dep} = \s
Codeforces 1009F 长链剖分优化dp做法
stdforces的博客
09-19 274
长链剖分优化dp
洛谷P5904 树形dp，长链剖分优化
stdforces的博客
09-20 276
树形dp
【2019.1.8模拟赛】Tree 【长链剖分】【凸优化】【DP】
Never give in.
01-10 273
Description 给出一棵n个点的树，每个点有点权，要求在树上选出恰好m条点不相交的链，每条链至少有k个点，要求点权和最大。 n≤150005n\leq 150005n≤150005 Solution 看到恰好选出m条链，立刻反应到是凸优化，二分选择一条链需要花费的额外代价 但是要求每条链至少有K个点。 设f[i]f[i]f[i]为i子树中随便选若干条链的答案（满足k的限制） 一个点的贡献大...
叶诗富教练详解树形动态规划： longest chain问题与应用
例如，给定一个树形结构，如树上最长链问题，目标是找到一条不重复的路径，其节点个数最多。这个问题涉及到了两个关键挑战：一是树的根节点不确定，需要确定根节点以便于动态规划；二是每个节点的最大子节点数可能...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值