Tree chain problem

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)

Total Submission(s): 262    Accepted Submission(s): 59

Problem Description
Coco has a tree, whose vertices are conveniently labeled by 1,2,…,n.

There are m chain on the tree, Each chain has a certain weight. Coco would like to pick out some chains any two of which do not share common vertices.

Find out the maximum sum of the weight Coco can pick
 
Input
The input consists of several test cases. The first line of input gives the number of test cases T (T<=10).

For each tests: 

First line two positive integers n, m.(1<=n,m<=100000)

The following (n - 1) lines contain 2 integers ai bi denoting an edge between vertices ai and bi (1≤ai,bi≤n),

Next m lines each three numbers u, v and val(1≤u,v≤n,0<val<1000), represent the two end points and the weight of a tree chain.
 
Output
For each tests:

A single integer, the maximum number of paths.
 
Sample Input
1
7 3
1 2
1 3
2 4
2 5
3 6
3 7
2 3 4
4 5 3
6 7 3
 
Sample Output
6
Hint
Stack expansion program: #pragma comment(linker, "/STACK:1024000000,1024000000")

题目大意:有一颗n个节点的数。给出n-1条边(无向),还有m条链,每条链链接两个顶点(按lca的方式链接)。链存在一个权值w。如今想要挑选一些链,挑选的链中不能出现同样的节点,问能够挑选出的最大的权重是多少?

要求权值最大,依照树形dp的思路去考虑,那么dp[i]为以第i个点位根节点的子树的最优解。dp[i]的状态转移公式,有两种可能,第一种:第i个节点上不出现链,那么dp[i] = ∑(dp[k] | k为i的子节点);另外一种:第i个节点上出现链。假设选择增加这条链,那么dp[i] = w(链的权值) + ∑(dp[k] | k为链上的节点的子节点) = w + ∑(sum[k]
| k为链上的节点 ) - ∑(dp[k] | k为链上的节点) 。sum[i]表示i节点的全部子节点的dp和,在 ∑(sum[k] | k为链上的节点 ) - ∑(dp[k] | k为链上的节点) 中减去的dp[k]会由它的父节点的sum补全。

这样就得到了状态转移公式。

还有要计算给出的链的两个顶点的lca(用于更新dp的值),在更新dp值的时候须要计算链上节点的(sun的和)(dp的和),使用树状数组来计算。对全部节点进行dfs。对点进行又一次编号。每到达一个点和每离开一个点都要编号,记录在l[i]和r[i]中,当计算玩一个sum[i]值后。加到树状数组c1中(l[i]位置加正值,r[i]位置加负值)。计算完一个dp[i]值后,加到树状数组c2中(l[i]位置加正值,r[i]位置加负值)。依照新的编号进行树状数组计算区间和。也就能够得到链上的节点的和

总结:

1、建树,题目给出的(无向边)。要加双边

2、dfs,建立rmq[i][j]:i节点的第2^j个父节点,对节点又一次编号

3、对给出的两个顶点求lca。(代码中用rmq求lca)

4、树状数组,求区间和。

5、深搜,完毕树形dp,按状态转移方程计算每个节点的值,使用树状数组计算链上节点的和

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std ;
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define maxn 200050
struct E{
int v , next ;
}edge[maxn] ;
int head[maxn] , cnt ;
struct node{
int u , v , w ;
int lca ;
}p[maxn] ;
int dep[maxn] , rmq[maxn][20] ;
int l[100010] , r[maxn] , cid ;
vector <int> vec[maxn] ;
int dp[maxn] , sum[maxn] ;
int c1[maxn] , c2[maxn] ;
void add_E(int u,int v) {
edge[cnt].v = v ;
edge[cnt].next = head[u] ; head[u] = cnt++ ;
edge[cnt].v = u ;
edge[cnt].next = head[v] ; head[v] = cnt++ ;
}
void dfs(int fa,int u) {
l[u] = ++cid ;
int i , j , v ;
for(i = head[u] ; i != -1 ; i = edge[i].next) {
v = edge[i].v ;
if( v == fa ) continue ;
dep[v] = dep[u] + 1 ;
rmq[v][0] = u ;
for(j = 1 ; j < 20 ; j++)
rmq[v][j] = rmq[ rmq[v][j-1] ][j-1] ;
dfs(u,v) ;
}
r[u] = ++cid ;
}
int lca(int u,int v) {
if( dep[u] < dep[v] )
swap(u,v) ;
int i , j ;
for(i = 19 ; i >= 0 ; i--) {
if( dep[ rmq[u][i] ] >= dep[v] )
u = rmq[u][i] ;
if( u == v ) return u ;
}
for(i = 19 ; i >= 0 ; i--) {
if( rmq[u][i] != rmq[v][i] ){
u = rmq[u][i] ;
v = rmq[v][i] ;
}
}
return rmq[u][0] ;
}
int lowbit(int x) {
return x & -x ;
}
void add(int i,int k,int *c,int n) {
while( i <= n ) {
c[i] += k ;
i += lowbit(i) ;
}
}
int getsum(int i,int *c) {
int ans = 0 ;
while( i ) {
ans += c[i] ;
i -= lowbit(i) ;
}
return ans ;
}
void solve(int fa,int s,int n) {
dp[s] = sum[s] = 0 ;
int i , j , u , v , temp ;
for(i = head[s] ; i != -1 ;i = edge[i].next) {
v = edge[i].v ;
if( v == fa ) continue ;
solve(s,v,n) ;
sum[s] += dp[v] ;
}
dp[s] = sum[s] ;
for(i = 0 ; i < vec[s].size() ; i++) {
u = p[ vec[s][i] ].u ;
v = p[ vec[s][i] ].v ;
temp = getsum(l[u],c1) + getsum(l[v],c1) - getsum(l[u],c2) - getsum(l[v],c2) + sum[s] ;
dp[s] = max(dp[s],temp+p[vec[s][i]].w) ;
}
add(l[s],sum[s],c1,n*2) ;
add(r[s],-sum[s],c1,n*2) ;
add(l[s],dp[s],c2,n*2) ;
add(r[s],-dp[s],c2,n*2) ;
}
void init(int n) {
memset(head,-1,sizeof(head)) ;
memset(rmq,0,sizeof(rmq)) ;
memset(c1,0,sizeof(c1)) ;
memset(c2,0,sizeof(c2)) ;
cnt = cid = 0 ;
dep[1] = 1 ;
rmq[1][0] = 1 ;
for(int i = 1 ; i <= n ; i++)
vec[i].clear() ;
return ;
}
int main() {
int t , n , m ;
int i , j , u , v , w ;
//freopen("1006.in","r",stdin) ;
//freopen("t.out","w",stdout) ;
scanf("%d", &t) ;
while( t-- ) {
scanf("%d %d", &n, &m) ;
init(n) ;
for(i = 1 ; i < n ; i++) {
scanf("%d %d", &u, &v) ;
add_E(u,v) ;
}
dfs(-1,1) ;
for(i = 0 ; i < m ; i++) {
scanf("%d %d %d", &p[i].u, &p[i].v, &p[i].w) ;
p[i].lca = lca(p[i].u,p[i].v) ;
vec[ p[i].lca ].push_back(i) ;
}
solve(-1,1,n) ;
printf("%d\n", dp[1]) ;
}
return 0 ;
}

hdu5293(2015多校1)--Tree chain problem(树状dp)的更多相关文章

  1. 【HDU 5233】Tree chain problem (树形DP+树剖+线段树|树状数组)最大权不相交树链集

    [题目] Tree chain problem Problem Description Coco has a tree, whose vertices are conveniently labeled ...

  2. [HDU 5293]Tree chain problem(树形dp+树链剖分)

    [HDU 5293]Tree chain problem(树形dp+树链剖分) 题面 在一棵树中,给出若干条链和链的权值,求选取不相交的链使得权值和最大. 分析 考虑树形dp,dp[x]表示以x为子树 ...

  3. poj2486--Apple Tree(树状dp)

    Apple Tree Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 7789   Accepted: 2606 Descri ...

  4. 刷题总结——Tree chain problem(HDU 5293 树形dp+dfs序+树状数组)

    题目: Problem Description Coco has a tree, whose vertices are conveniently labeled by 1,2,…,n.There ar ...

  5. 树形DP+DFS序+树状数组 HDOJ 5293 Tree chain problem(树链问题)

    题目链接 题意: 有n个点的一棵树.其中树上有m条已知的链,每条链有一个权值.从中选出任意个不相交的链使得链的权值和最大. 思路: 树形DP.设dp[i]表示i的子树下的最优权值和,sum[i]表示不 ...

  6. Codeforces Round #225 (Div. 1) C. Propagating tree dfs序+树状数组

    C. Propagating tree Time Limit: 20 Sec Memory Limit: 256 MB 题目连接 http://codeforces.com/contest/383/p ...

  7. Codeforces Round #225 (Div. 1) C. Propagating tree dfs序+ 树状数组或线段树

    C. Propagating tree Time Limit: 20 Sec Memory Limit: 256 MB 题目连接 http://codeforces.com/contest/383/p ...

  8. [poj3321]Apple Tree(dfs序+树状数组)

    Apple Tree Time Limit: 2000MS   Memory Limit: 65536K Total Submissions: 26762   Accepted: 7947 Descr ...

  9. POJ3321Apple Tree Dfs序 树状数组

    出自——博客园-zhouzhendong ~去博客园看该题解~ 题目 POJ3321 Apple Tree 题意概括 有一颗01树,以结点1为树根,一开始所有的结点权值都是1,有两种操作: 1.改变其 ...

随机推荐

  1. msyql安装

    1.安装msyql yum install -y ncurses-devel automake autoconf bison libtool-ltdl-devel cd /soft wget http ...

  2. es6字符串模板总结

    我们平时用原生js插入标签或者用node.js写数据库语言时候,经常需要大量的字符串进行转义,很容易出错,有了es6的字符串模板,就再也不用担心会出错了 1.模板中的变量写在${}中,${}中的值可以 ...

  3. TensorFlow-GPU安装配置(win10+tensorflow1.6+CUDA9.0+cudnn7.0+python3.6+Visual Studio2013)

    安装步骤: TensorFlow官网 tensorflow一般只能装在python3上,CUDA9.0搭配cudnn7.0,CUDA8.0搭配cudnn6.0 查看对应要安装的环境版本(因为会不断更新 ...

  4. POJ 3657 Haybale Guessing(区间染色 并查集)

    Haybale Guessing Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 2384   Accepted: 645 D ...

  5. 挑战python 之一马当先(python的广搜)

    下过象棋的人都知道,马只能走'日'字形(包括旋转90°的日),现在想象一下,给你一个n行m列网格棋盘, 棋盘的左下角有一匹马,请你计算至少需要几步可以将它移动到棋盘的右上角,若无法走到,则输出-1. ...

  6. 【BZOJ 3669】 3669: [Noi2014]魔法森林 (动态spfa)

    3669: [Noi2014]魔法森林 Description 为了得到书法大家的真传,小E同学下定决心去拜访住在魔法森林中的隐士.魔法森林可以被看成一个包含个N节点M条边的无向图,节点标号为1..N ...

  7. 【树形dp】Find Metal Mineral

    [HDU4003]Find Metal Mineral Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65768/65768 K (J ...

  8. Uncaught SyntaxError: Invalid shorthand property initializer

    $.ajax({ url : '../../collateralQuery/getCollateralQueryDetail', type : 'POST', data : {}, dataType ...

  9. Generator函数(三)

    Generator.prototype.return() 1.Generator函数返回的遍历器对象还有一个return方法,可以返回给定的值,并终结Generator函数的遍历. function* ...

  10. css3背景属性 background-size 对背景图进行缩小放大

    background-size需要两个值,它的类型可以是像素(px).百分比(%)或是auto,还可以是cover和contain.第一个值为背景图的width,另外一个值用于指定背景图上的heigh ...