视频
UP主的博客
- 暴力做法,能过 42%数据。如果内存开 1e4+10 能过 40%,如果开 2e5+10就只能过 25%
#include<bits/stdc++.h>
#define int long long
#define endl '\n'
const int N = 1e4+10;
using namespace std;
//存两点的距离
typedef pair<int,int> PII;
map<PII,int> st;
//存游览路径
int a[N];
//存图
vector<PII> edge[N];/*u:起点v:终点n:当前节点p:当前节点的父节点sum:u 到 n 的路径长dfs:搜索从 u 到 v 的路径长度,记录在 st 中*/
bool dfs(int u,int v,int n,int p,int sum){if(n==v){st[{u,v}]=sum;st[{v,u}]=sum;return true;}for(int i=0;i<edge[n].size();i++){int son = edge[n][i].first;if(son == p)continue;int w = edge[n][i].second;if(dfs(u,v,son,n,sum+w))return true;}return false;
}void solve(){int n,k;cin>>n>>k;for(int i=1;i<=n-1;i++){int u,v,w;cin>>u>>v>>w;edge[u].push_back({v,w});edge[v].push_back({u,w});}for(int i=1;i<=k;i++)cin>>a[i];int ans = 0;for(int i=1;i<=k-1;i++){dfs(a[i],a[i+1],a[i],-1,0);ans += st[{a[i],a[i+1]}];}for(int i=1;i<=k;i++){int tmp = ans;if(i==1){tmp-=st[{a[i],a[i+1]}];}else if(i==k){tmp-=st[{a[i-1],a[i]}];}else{tmp-=st[{a[i-1],a[i]}];tmp-=st[{a[i],a[i+1]}];dfs(a[i-1],a[i+1],a[i-1],-1,0);tmp+=st[{a[i-1],a[i+1]}];}cout<<tmp<<endl;}}signed main( ){ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);int T;T = 1;while (T--)solve( );return 0;
}
- 正解:最近公共祖先
- LCA
- tarjan
- 树链剖分(博主觉得更好写)
这道题听说是最近公共祖先的模板题,会模板就能做。