讨伐[天梯赛训练]

题目描述

输入样例
4 2
1 3 5 7
1 2 2
3 4 3
输出样例
8

思路

最短路 + 位运算

存储结构
1.使用邻接表存储每个城市节点的连通性及其距离
2.一维数组dis存储起点到每个点的距离，如dis[i]表示起点到i的距离
3.bool数组vis标记每个点是否被访问过，防止重复寻找起点到i点的最短路径

前导知识
lowbit函数可以计算n的二进制表示中有多少个1

原理
①x & -x 通常用于计算一个整数 x 的最低有效位
②由于 -x 是 x 的补码表示的负数，其特点是除了最低有效位以外，其他位与 x 相同，最低有效位则从0变为了1。因此，x & -x 的结果就是将 x 的最低有效位变为0，其他位保持不变

#include<iostream>
using namespace std;
int lowbit(int x)//返回的是一个二进制数，如100100，第一次返回的就是100，即8
{
    return x&-x;
}
int main()
{
    int x;
    cin>>x;
    int ans=0;
    while(x)
    {
        x-=lowbit(x);
        ans++;
    }
    cout<<ans<<endl;
    return 0;
}

具体做法
1.对于每一个点，其权值对应的二进制数1的个数为x，由于题目中给出节点的编号范围是1~2e5，因此设置一个偏移量2e5，建立虚拟节点（范围为2e5 + 1 到 2e5 + 30，因为权值最大为1e9，在这个范围内，二进制数表示中最多有30个1
2.建立i点到虚拟节点的单向路径，因为虚拟节点并不存在，所以是单向路径，否则会将本不存在的路径加入最短路径中
3.建立m条确定存在的双向路
4.使用堆优化版的dijkstra算法寻找最短路径(n和m同数量级，为稀疏图)

注意
1.dis数组要开到long long，因为最大为1e9，多条加起来会超过int范围
2.long long 对应的最大值为0x3f3f3f3f3f3f3f3f

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, int> PII;
const int N = 1000010, C = 2e5;
const ll INF = 0x3f3f3f3f3f3f3f3f;
int h[N], e[N], ne[N], w[N], idx;
ll dis[N]; //dis[i]表示从1点到i点的距离
bool vis[N]; //标记每个点是否被访问过
void add(int a, int b, int c)
{
    w[idx] = c;
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

void dijkstra()
{
    memset(dis, 0x3f, sizeof(dis));
    dis[1] = 0;
    priority_queue<PII, vector<PII>, greater<PII>> he; //小根堆
    he.push({0, 1});
    while(!he.empty())
    {
        PII t = he.top();
        he.pop();
        int pos = t.second;
        if(vis[pos]) continue;
        vis[pos] = true;
        for(int i = h[pos]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dis[j] > dis[pos] + w[i])
            {
                dis[j] = dis[pos] + w[i];
                he.push({dis[j], j});
            }
        }
    }
}

int lowbit(int x)
{
    return x&-x;
}
int main()
{
    memset(h, -1, sizeof(h));
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i ++)
    {
        int x;
        cin >> x;
        int temp = x, sum = 0;
        while(temp)
        {
            temp -= lowbit(temp);
            sum ++;
        }
        add(i, C + sum, x);
        add(C + sum, i, 0);
    }
    while(m --)
    {
        int u, v, w;
        cin >> u >> v >> w;
        add(u, v, w);
        add(v, u, w);
    }
    dijkstra();
    if(dis[n] >= INF) cout << "gou yun hao" << endl;
    else cout << dis[n] << endl;
    return 0;
}

欢迎大家批评指正！！！