第4届蓝桥杯--大臣的旅费C++

 首先,要存储图,由于该图是稀疏图,所以用邻接表存储图

题目要求找出两个点,使其路程的花费最多,对于每个点,我们只需找到离这个点最远和次远的两个点,他们之间的路径就是最远路径,直接用dfs暴搜,如下:

int dfs(int u){
    st[u]=true;
    int d1=0,d2=0;//分别存储最远和次远路径
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        if(!st[j]){//如果该点没有被搜过
            int s=dfs(j)+w[i];
            if(s>=d1)d2=d1,d1=s;
            else if(s>d2)d2=s;
        }
    }
    ans=max(ans,d1+d2);//更新答案
    return d1;//返回最长路径
}

搜的时候,以任意一个点为起点都行,我这里写的dfs(1),从一号点开始搜 

在找到最远路径ans之后,路程的花费就是ans*10+(1+2+...+ans)=ans*10+ans*(1+ans)/2;

由于答案可能会很大,就输出ans*10+(long long)ans*(ans+1)/2即可

C++代码如下:

#include<iostream>
#include<cstring>
using namespace std;
const int N=100010,M=N*2;
typedef long long LL;
int h[N],e[M],w[M],ne[M],idx;
bool st[N];
int n,ans;
void add(int a,int b,int c){//加一条a->b的边,长度为c
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u){
    st[u]=true;//标记当前点已经被搜过
    int d1=0,d2=0;//分别存储最大和次大距离
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        if(!st[j]){//如果当前点没有被搜过
            int s=dfs(j)+w[i];
            if(s>=d1)d2=d1,d1=s;
            else if(s>d2)d2=s;
        }
    }
    ans=max(ans,d1+d2);//更新ans
    return d1;
}
int main(){
    cin>>n;
    memset(h,-1,sizeof h);//初始化
    for(int i=0;i<n-1;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c),add(b,a,c);//双向边
    }
    dfs(1);
    cout<<ans*10+(LL)ans*(ans+1)/2<<endl;
    return 0;
}

  • 11
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值