蓝桥杯练习：景区导游

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

• 正解：最近公共祖先

1. LCA
2. tarjan
3. 树链剖分（博主觉得更好写）
   这道题听说是最近公共祖先的模板题，会模板就能做。