标题必须大于五个字[doge]
树形dp就是将树的某一个节点当作根节点挂起来进行dp
树的最长路径模板
//树型dp求解带权树的最大直径
//先将树挂起来,易知,树的直径一定产生于某个节点的两个向下最长边
#include<bits/stdc++.h>
using namespace std;
#define N 100100
#define M 200100
int h[N],to[M],ne[M],w[M];
//其中 M长度的数组模拟一个结构体链式前向星,ne存储下一个节点的位置
//to存储这条边连接的点
int cnt=0;
void add( int a,int b,int c){
w[cnt]=c;to[cnt]=b;
ne[cnt]=h[a];
h[a]=cnt++;
return ;
}
//上面是链式前向星,将每个新申请的节点插在头节点和第一个之间
int ans=0;
int dfs(int rt,int fa){//本函数返回从rt节点向下的最大长度
int dist=0;
int d1=0,d2=0;
for( int i=h[rt];i!=-1;i=ne[i]){
if(to[i]==fa) continue;//注意不要向上dp
int d=w[i]+dfs(to[i],rt);
dist=max(dist,d);
if(d>=d1) d2=d1,d1=d;
else if(d>d2) d2=d;
}
ans=max(ans,d1+d2);
return dist;
}
int main(){
int n;
cin>>n;
memset(h,-1,sizeof(h));
for( int i=1;i<=n-1;i++){
int x,y,z;
cin>>x>>y>>z;
add(x,y,z);
add(y,x,z);
}
dfs(1,-1);
cout<<ans;
}
树的中心
树的中心:
树的中心定义是树中的某一个节点到其他点的最远距离最小。
首先要求解某个点到其他所有点的最短距离:
求解方法是先将树按某一个节点挂起来,然后从所求节点分别向上和向下搜索。
其中,向下走的最大长度用“树的直径求法”。
向上走的求法:求不走会原节点的路径的最大值。(不走回原节点的路径又可以分为两大类,一类是向上,另一类是向下,只需递归地求解即可)
所以求解树的中心的步骤是:
首先dfs求出每个节点向下走的最大值和次大值。
然后求解每个节点向上走的最大值。
向上走的最大值是max(再向上走的最大值,向下走且不经过原节点的最大值)
#include<bits/stdc++.h>
using namespace std;
#define N 10010
#define M 20010
int w[M],to[M],ne[M],h[N];
int cnt;
int dist1[N];//每个点向下走的最长路径
int from[N];//记录最大值的产生节点
int dist2[N];//每个点向下走的经过不同点的次长路径
int ans=0x7fffffff;
void add( int a,int b,int c){
w[cnt]=c;
to[cnt]=b;
ne[cnt]=h[a];
h[a]=cnt++;
return ;
}
int dfs(int fa,int tr){
int dist=0;//返回值
int d1=0,d2=0;//分别表示最大值和次大值
int mp=0;
for( int i=h[tr];i!=-1;i=ne[i]){
int j=to[i];
if(j==fa) continue;
int now=dfs(tr,j)+w[i];
dist=max(dist,now);
if(now>=d1) {
d2=d1,d1=now;
from[tr]=to[i];
}
else if(now>d2) d2=now;
}
dist1[tr]=d1;dist2[tr]=d2;
return dist;
}
void dfs2( int fa,int rt,int up){
ans=min(ans,max(up,dist1[rt]));
// cout<<ans<<" "<<up<<" "<<dist1[rt]<<" "<<rt<<endl;
for( int i=h[rt];i!=-1;i=ne[i]){
int j=to[i];
if(j==fa) continue;
int now=0;//now代表rt的子节点向上走的最大值
//以下是寻找子节点rt的向上最大值
if(from[rt]!=j) now=dist1[rt];
else now=dist2[rt];
now=max(up,now);
dfs2(rt,j,now+w[i]);
}
}
int main(){
int n;
cin>>n;
memset(h,-1,sizeof(h));//***
for( int i=1;i<=n-1;i++){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
dfs(-1,1);
dfs2(-1,1,0);
cout<<ans;
}
acwing1075/《信息学奥赛一本通》
数字转换
如果把每个数的因子和看作是父节点,那么每个数有且只有一个父节点,因此整个数据结构是一颗树
#include<bits/stdc++.h>
using namespace std;
#define N 50010
#define M 100010
int ne[M],to[M],h[N];
int cnt,n;
int vis[N];
void add(int a,int b){
to[cnt]=b;
ne[cnt]=h[a];
h[a]=cnt++;
return ;
}
void build(){//这里是用试除法求出每个数的因子
for( int i=1;i<=n;i++){
int sum=1;
for( int j=2;j*j<=i;j++){
if(i%j==0&&j*j!=i){
sum+=j;
sum+=i/j;
}
else if(j*j==i) sum+=j;
}
if(sum<i){
// cout<<i<<" "<<sum<<endl;
add(i,sum);
add(sum,i);
}
}
}
int sum[N];//记录每个数的因子和
void build2(){//这里用筛法求每个数的因子,时间复杂度是lnn
for( int i=1;i<=n;i++){
for( int j=i*2;j<=n;j+=i){
sum[j]+=i;
}
}
for( int i=1;i<=n;i++){
if(i>sum[i]){
add(i,sum[i]);
add(sum[i],i);
}
}
}
int ans=0;
int dfs(int rt,int fa){
int dist=0;
int d1=0,d2=0;
for( int i=h[rt];i!=-1;i=ne[i]){
int j=to[i];
if(j==fa) continue;
int now=dfs(j,rt)+1;
dist=max(dist,now);
if(now>=d1) d2=d1,d1=now;
else if(now>d2) d2=now;
}
ans=max(ans,d1+d2);
// cout<<rt<<" "<<d1+d2<<endl;
return dist;
}
int main(){
cin>>n;
memset(h,-1,sizeof(h));
build();
dfs(1,-1);
cout<<ans;
}
因为本题的边没有权值,因此 ,本题也可以使用找树的直径的经典做法,就是先dfs找到最远点,再从最远点dfs再找最远点
#include<bits/stdc++.h>
using namespace std;
#define N 50010
#define M 100010
int ne[M],to[M],h[N];
int cnt,n;
int vis[N];
void add(int a,int b){
to[cnt]=b;
ne[cnt]=h[a];
h[a]=cnt++;
return ;
}
int sum[N];//记录每个数的因子和
void build2(){//这里用筛法求每个数的因子,时间复杂度是lnn
for( int i=1;i<=n;i++){
for( int j=i*2;j<=n;j+=i){
sum[j]+=i;
}
}
for( int i=1;i<=n;i++){
if(i>sum[i]){
add(i,sum[i]);
add(sum[i],i);
}
}
}
int point;
int maxx=0;
void dfs( int step,int fa,int rt){
if(step>maxx){
maxx=step;
point=rt;
}
for( int i=h[rt];i!=-1;i=ne[i]){
int j=to[i];
if(j==fa) continue;
dfs(step+1,rt,j);
}
}
int main(){
cin>>n;
memset(h,-1,sizeof(h));
build2();
dfs(1,1,-1);
maxx=0;
dfs(0,-1,point);
cout<<maxx;
}