牛客补题(BFS+DP)

最新推荐文章于 2024-11-07 21:17:23 发布

HZ35572

最新推荐文章于 2024-11-07 21:17:23 发布

阅读量300

点赞数 1

分类专栏：题解文章标签：算法

本文链接：https://blog.youkuaiyun.com/qq_51699436/article/details/120617124

版权

题解专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用广度优先搜索（BFS）算法解决无向图中从特定节点出发到达w值不小于指定值的最近节点的最短路径问题。通过构建无权值为1的边的图，对于每个节点的w值，利用BFS进行打表计算，并存储每个节点到所有w值不超过当前值的节点的最短距离。最后，根据输入的查询，快速返回从起点到目标w值节点的最短路径长度。代码实现了该算法并展示了样例输入和输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Spicy Restaurant

题意
一个有n个节点m条边的无向图，每条边权值为1，每个节点有一个w值，有q次询问，每次询问从p点出发到达最近的一个w值不小于a的点的距离
思路：用bfs和dp进行打表，输入输出要用scanf和printf，不然会tle
代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<cmath>
#include<map>
#include<unordered_map>
#include<unordered_set>
#include<stack>
#define N 100005
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> P;

int t,n,m,q,dis[105][N],que[N*100];
vector<int> w[N];
vector<int> G[N];

void bfs(vector<int> &v,int x,int *dis){
    int head=0,tail=0;
    for(int i=1;i<=n;i++)
        dis[i]=inf;
    for(auto xx:v){
        que[++tail]=xx;
        dis[xx]=0;
    }
    while(tail!=head){
        int qq=que[++head];
        for(auto to:G[qq]){
            if(dis[to]==inf){
                dis[to]=dis[qq]+1;
                que[++tail]=to;
            }
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&q);
    int W,x,y;
    for(int i=1;i<=n;i++){
        scanf("%d",&W);
        w[W].push_back(i);
    }

    for(int i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }

    for(int i=1;i<=100;i++){
        bfs(w[i],i,dis[i]);
    }

    for(int i=2;i<=100;i++){
        for(int j=1;j<=n;j++){
        	//dis[i][j]表示从j点出发,到达一个w值不超过i的最短距离
            dis[i][j]=min(dis[i][j],dis[i-1][j]);
        }
    }

    int p,a;
    for(int i=1;i<=q;i++){
        scanf("%d%d",&p,&a);
        printf("%d\n",dis[a][p]==inf?-1:dis[a][p]);
    }
    return 0;
}

/*
4 4 5
5 4 2 3
1 2
2 3
3 4
4 1
1 1
1 2
1 3
1 4
1 5
*/