蓝桥杯刷题-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;
}
相关推荐
desssq5 小时前
力扣:70. 爬楼梯
算法·leetcode·职场和发展
June bug1 天前
【软考中级·软件评测师】下午题·面向对象测试之架构考点全析:分层、分布式、微内核与事件驱动
经验分享·分布式·职场和发展·架构·学习方法·测试·软考
薰衣草23331 天前
一天两道力扣(1)
算法·leetcode·职场和发展
爱coding的橙子1 天前
每日算法刷题Day41 6.28:leetcode前缀和2道题,用时1h20min(要加快)
算法·leetcode·职场和发展
一只鱼^_1 天前
基础算法合集-图论
数据结构·算法·深度优先·图论·广度优先·宽度优先·图搜索算法
天真小巫2 天前
2025.6.27总结
职场和发展
Memories off11 天前
实习/秋招记录:软件开发转AI或安全
职场和发展
一入JAVA毁终身11 天前
针对我的简历模拟面试
面试·职场和发展
挑战者66688811 天前
如何制定团队制度?
职场和发展·项目管理·创业创新
稳兽龙11 天前
P3258 [JLOI2014] 松鼠的新家
数据结构·c++·算法·深度优先·lca