蓝桥杯刷题-04-岛屿个数-DFS


cpp 复制代码
#include <iostream>
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10;
typedef pair<int,int>pii;

map<pii, int>st;//记录从{x,y}的距离是多少
int a[N];//存储原始路径


vector<pii>edge[N];//存图

// s表示你要求的路径的起点
// v表示你要求的路径的终点
//u表示你当前走到了哪个点
//father表示你当前这个点的父亲节点是谁。避免重复走造成死循环。
// sum表示从s走到u的路径花费总和。
bool dfs(int s,int u,int father,int v,int sum)
{
  if(u==v){
    st[{s,v}]=sum;
    st[{v,s}]=sum;
    return true;
  }
  for(int i=0;i<edge[u].size();i++)
  {
    int son=edge[u][i].first;
    if(son==father)
      continue;
    int w=edge[u][i].second;

    if(dfs(s,son,u,v,sum+w))
      return true;

  }
  return false;
}


void sovle()
{
  int n,k;
  cin>>n>>k;
  for(int i=0;i<n-1;i++)
  {
    int x,y,t;
    cin>>x>>y>>t;
    edge[x].push_back({y,t});
    edge[y].push_back({x,t});
  }
  for(int i=0;i<k;i++)
    cin>>a[i];
  
  int ans=0;

  //总花费
  for(int i=0;i<k-1;i++)
  {
    dfs(a[i],a[i],-1,a[i+1],0);
    ans+=st[{a[i],a[i+1]}];
  }

  //跳过某个景点的花费
  for(int i=0;i<k;i++)
  {
    int tmp=ans;
    if(i==0)
      tmp-=st[{a[i],a[i+1]}];
    else if(i==k-1)
      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],-1,a[i+1],0);
      tmp+=st[{a[i-1],a[i+1]}];

    }
    cout<<tmp<<" ";
  }
}


signed main()
{
  // DFS遍历
  ios::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  int t=1;
  //cin>>t;
  while(t--)
  sovle();

  return 0;
}
相关推荐
lally.7 分钟前
2025蓝桥杯省赛网络安全组wp
web安全·蓝桥杯
我想进大厂2 小时前
图论---最大流(Dinic)
算法·深度优先·图论
zhilanguifang5 小时前
面试记录1-春招补录0427
面试·职场和发展
WebGIS小智5 小时前
WebGIS面试题目整合资料
面试·职场和发展
东方翱翔6 小时前
第十六届蓝桥杯大赛软件赛省赛第二场 C/C++ 大学 A 组
算法·职场和发展·蓝桥杯
Y1nhl7 小时前
力扣hot100_子串_python版本
开发语言·python·算法·leetcode·职场和发展
wuqingshun3141598 小时前
蓝桥杯 16. 密文搜索
c++·算法·职场和发展·蓝桥杯·深度优先
理智的灰太狼9 小时前
题目 3320: 蓝桥杯2025年第十六届省赛真题-产值调整
蓝桥杯
天真小巫20 小时前
2025.4.26总结
职场和发展
triticale20 小时前
【蓝桥杯】P12165 [蓝桥杯 2025 省 C/Java A] 最短距离
java·蓝桥杯