蓝桥杯练习：景区导游-编程知识

蓝桥杯练习：景区导游

视频
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;
}