蓝桥杯刷题-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;
}
相关推荐
不过四级不改名6773 小时前
蓝桥杯嵌入式备赛教程(1、led,2、lcd,3、key)
stm32·嵌入式硬件·蓝桥杯
TT哇3 小时前
*【每日一题 提高题】[蓝桥杯 2022 国 A] 选素数
java·算法·蓝桥杯
chengooooooo4 小时前
代码随想录训练营第二十七天| 贪心理论基础 455.分发饼干 376. 摆动序列 53. 最大子序和
算法·leetcode·职场和发展
测试杂货铺6 小时前
如何用postman做接口自动化测试及完美的可视化报告?
自动化测试·软件测试·测试工具·职场和发展·jenkins·压力测试·postman
TT哇7 小时前
*【每日一题 基础题】 [蓝桥杯 2023 省 B] 飞机降落
算法·蓝桥杯
撞上电子7 小时前
蓝桥杯物联网开发板硬件组成
物联网·职场和发展·蓝桥杯
pzx_0017 小时前
【LeetCode】LCR 175.计算二叉树的深度
开发语言·c++·算法·leetcode·职场和发展
Aloha_up7 小时前
LeetCode hot100-89
算法·leetcode·职场和发展
测试老哥10 小时前
Jmeter测试脚本编写技巧
自动化测试·软件测试·功能测试·测试工具·jmeter·职场和发展·性能测试
逝灮11 小时前
【蓝桥杯——物联网设计与开发】拓展模块4 - 脉冲模块
stm32·单片机·嵌入式硬件·mcu·物联网·蓝桥杯·脉冲测量