树的最长路径
思路 每次都把每个点看成根节点之后进行向下进行遍历每次求最大 和次大值把他相加 不断进行搜索
#include <bits/stdc++.h>
using namespace std;
#define FAST ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
// define int long long
typedef long long LL;
typedef pair<int, int> PII;
const int N = 2e5 + 10;
int n, m, a[N];
int h[N], e[N], ne[N], w[N], idx;
int ans;
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u, int father)
{
int dis = 0, d1 = 0, d2 = 0;
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if (j == father)
continue;
int d = dfs(j, u) + w[i];
dis = max(dis, d);
if (d >= d1)
d2 = d1, d1 = d;
else if (d > d2)
d2 = d;
}
ans = max(ans, d1 + d2);
return dis;
}
void solve()
{
cin >> n;
memset(h,-1,sizeof(h));
for (int i = 1; i <= n - 1; i++)
{
int a, b, c;
cin >> a >> b >> c;
add(a, b, c), add(b, a, c);
}
dfs(1, -1);
cout << ans << endl;
}
signed main()
{
FAST;
int T = 1;
// cin>>T;
while (T--)
{
solve();
}
}
分析
树的中心
换根dp:
1 指定任意一个为子节点 就是全部更新
2一次dfs统计出档期子树的结点对当前结点的贡献度
3从当前节点向上走找到父节点 父节点出发的且不回到该节点的贡献度 统计答案;
#include <bits/stdc++.h>
using namespace std;
#define FAST ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
// #define int long long
typedef long long LL;
typedef pair<int, int> PII;
const int N = 2e5 + 10, INF = 0x3f3f3f3f;
int n, m, a[N];
int p[N], d1[N], d2[N];
int e[N], w[N], idx, ne[N], h[N], up[N];
bool is_ye[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
int dfs_d(int u, int father)//向下找每个节点的项下的最大值和次大值
{
d1[u] = d2[u] = -INF;
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if (j == father)
continue;
int d = dfs_d(j, u) + w[i];//深搜变值
if (d >= d1[u])
{
d2[u] = d1[u], d1[u] = d;
p[u] = j;
}
else if (d > d2[u])
d2[u] = d;
}
if (d1[u] ==-INF)//叶子结点改变
{
d1[u] = d2[u] = 0;
is_ye[u] = true;
}
return d1[u];
}
void dfs_u(int u, int father)//向上找最值(相当于向上之后再找向下那种)
{
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if (j == father)
continue;
if (p[u] == j)
up[j] = max(up[u], d2[u]) + w[i];//如果最大值又3回去了 则用次大值
else
up[j] = max(up[u], d1[u]) + w[i];//没走过则用就可以
dfs_u(j,u);//别忘了搜索下一层
}
}
void solve()
{
cin >> n;
memset(h, -1, sizeof(h));
for (int i = 0; i < n - 1; i++)
{
int a, b, c;
cin >> a >> b >> c;
add(a, b, c), add(b, a, c);
}
dfs_d(1, -1);
dfs_u(1, -1);
int ans = d1[1];
for (int i = 2; i <= n; i++)
{
if (is_ye[i])
ans = min(ans, up[i]);
else
ans = min(ans, max(up[i], d1[i]));
}
cout << ans << endl;
}
signed main()
{
FAST;
int T = 1;
// cin>>T;
while (T--)
{
solve();
}
}
数字转换
有一个点入渡为一 之多一条边之后就是森林 (和树差不多)
还有一个点就是埃式筛法 求和 1-n之间的约数之和;
#include <bits/stdc++.h>
using namespace std;
#define FAST ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
// #define int long long
typedef long long LL;
typedef pair<int, int> PII;
const int N = 5e5 + 10;
int n, m, a[N];
int sum[N];
int ne[N], h[N], e[N], idx;
bool st[N];
int ans;
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u)//求树的最大直径
{
st[u] = true;
int dis = 0, d1 = 0, d2 = 0;
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if (!st[j])
{
int d = dfs(j);
dis = max(dis, d);
if (d >= d1)
d2 = d1, d1 = d;
else if (d > d2)
d2 = d;
}
}
ans = max(ans, d1 + d2);
return dis + 1;//返回其值
}
void solve()
{
cin >> n;
memset(h, -1, sizeof(h));
for (int i = 1; i <= n; i++)//埃式筛法 筛选出1-n的和
{
for (int j = 2; j <= n / i; j++)
sum[i * j] += i;
}
for (int i = 2; i <= n; i++)//插入树
if (sum[i] < i)
add(sum[i], i);
for (int i = 1; i <= n; i++)//遍历
{
if (!st[i])
dfs(i);
}
cout << ans << endl;
}
signed main()
{
FAST;
int T = 1;
// cin>>T;
while (T--)
{
solve();
}
}