c++新年好和通信路线（acwing）

第一个问题在于枚举

先看题目：

重庆城里有 n个车站，m 条 双向 公路连接其中的某些车站。

每两个车站最多用一条公路连接，从任何一个车站出发都可以经过一条或者多条公路到达其他车站，但不同的路径需要花费的时间可能不同。

在一条路径上花费的时间等于路径上所有公路需要的时间之和。

佳佳的家在车站 11，他有五个亲戚，分别住在车站 a,b,c,d,e。

过年了，他需要从自己的家出发，拜访每个亲戚（顺序任意），给他们送去节日的祝福。

怎样走，才需要最少的时间？

输入格式

第一行：包含两个整数 n,m，分别表示车站数目和公路数目。

第二行：包含五个整数 a,b,c,d,e，分别表示五个亲戚所在车站编号。

以下 m 行，每行三个整数 x,y,t，表示公路连接的两个车站编号和时间。

输出格式

输出仅一行，包含一个整数 T，表示最少的总时间。

数据范围

1≤n≤50000,
1≤m≤10^5,
1<a,b,c,d,e≤n,
1≤x,y≤n,
1≤t≤100

输入样例：

6 6
2 3 4 5 6
1 2 8
2 3 3
3 4 4
4 5 5
5 6 2
1 6 7

输出样例：

21

代码如下

#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>

using namespace std;

typedef pair<int, int> PII;
const int N = 5e4 + 10, M = 2e5 + 10, INF = 0x3f3f3f3f;

int n, m;
int h[N], w[M], e[M], ne[M], idx;
int source[N];
int dist[6][N];
bool st[N];

void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

void dijkstra(int start, int dist[])
{
    memset(dist, 0x3f, N * 4);//因为这边传过来的数组，实际是指针，所以是N*4
    memset(st, 0, sizeof st);
    
    dist[start] = 0;
    
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({0, start});
    
    while(heap.size())
    {
        auto t = heap.top();
        heap.pop();

        int ver = t.second;
        
        if(st[ver]) continue;
        st[ver] = true;
        
        for(int i = h[ver]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dist[j] > dist[ver] + w[i])
            {
                dist[j] = dist[ver] + w[i];
                heap.push({dist[j], j});
            }
        }
    }
}

// 枚举每种拜访次序，求出最小距离
// 拜访了u个人，自己是第1个人；当前起点是source[start]，当前走过的距离是distance
int dfs(int u, int start, int distance)
{
    if(u > 5) return distance;//走完了，返回
    
// res存距离最短的分支
    int res = INF;
    for(int i = 1; i < 6; i ++)
    {
        if(!st[i])
        {
            int next = source[i];//走亲戚
            st[i] = true;
            res = min(res, dfs(u + 1, i, distance + dist[start][next]));
            st[i] = false;
        }
    }
    
    return res;
}

int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    
    source[0] = 1;//第一个结点为1
    for(int i = 1;i <= 5; i ++) cin >> source[i];//输入剩余五个亲戚的站台号
    
    while(m --)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), add(b, a, c);
    }
    
    for(int i = 0; i < 6; i ++) dijkstra(source[i], dist[i]);//每一个节点都跑一次最短路
    
    memset(st, 0, sizeof st);//重置st数组，因为要再次使用
    
    cout << dfs(1, 0, 0) << endl;//输出答案
    
    return 0;
}

第二个题目：

博主用的分层图做法

先看题目：

在郊区有 N 座通信基站，P条 双向 电缆，第 i 条电缆连接基站 Ai 和 Bi。

特别地，11 号基站是通信公司的总站，N 号基站位于一座农场中。

现在，农场主希望对通信线路进行升级，其中升级第 i 条电缆需要花费 Li。

电话公司正在举行优惠活动。

农产主可以指定一条从 11 号基站到 N 号基站的路径，并指定路径上不超过 K 条电缆，由电话公司免费提供升级服务。

农场主只需要支付在该路径上剩余的电缆中，升级价格最贵的那条电缆的花费即可。

求至少用多少钱可以完成升级。

输入格式

第 11 行：三个整数 N，P，K。

第 2..P+1 行：第 i+1 行包含三个整数 Ai,Bi,Li。

输出格式

包含一个整数表示最少花费。

若 11 号基站与 N 号基站之间不存在路径，则输出 −1−1。

数据范围

0≤K<N≤1000,
1≤P≤10000,
1≤Li≤1000000

代码如下：

#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>

using namespace std;

typedef pair<int, int> PII;
//因为建议k层分层图，所以数据是这样的
const int N = 1e6 + 10, M = 2e7 + 10, INF = 0x3f3f3f3f;

int n, m, k;
int h[N], w[M], e[M], ne[M], idx;
int dist[N];
bool st[N];

void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

void dijkstra()
{
    memset(dist, 0x3f, sizeof dist);
    
    dist[1] = 0;
    
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({0, 1});
    
    while(heap.size())
    {
        auto t = heap.top();
        heap.pop();
        
        int ver = t.second, distance = t.first;
        
        if(st[ver]) continue;
        st[ver] = true;
        
        for(int i = h[ver]; ~ i; i = ne[i])
        {
            int j = e[i], x = max(dist[ver], w[i]);//这里是比较价格最大的那个
            if(dist[j] > x)
            {
                dist[j] = x;
                heap.push({dist[j], j});
            }
        }
    }
}

int main()
{
    cin >> n >> m >> k;
    
    memset(h, -1, sizeof h);
    
    while(m --)
    {
        int a, b, c;
        scanf("%d%d%d",&a, &b, &c);
        add(a, b, c), add(b, a, c);
        for(int j = 1; j <= k; j ++)//k次免费，建议k层分层图
        {
            add(j * n + a, j * n + b, c);
            add(j * n + b, j * n + a, c);
            add((j - 1) * n + a, j * n + b, 0);
            add((j - 1) * n + b, j * n + a, 0);
        }
    }
    
    //给每层终点建一条边是为了防止出现用不到k次就出现最短路的情况
    for(int i = 1; i <= k; i ++) add(i * n, (i + 1) * n, 0);
    
    dijkstra();
    
    if(dist[(k + 1) * n] == INF) puts("-1");
    else cout <<dist[(k + 1) * n] << endl;
    
    return 0;
}

明白的小伙伴麻烦点个赞吧