价格范围内最高排名的 K 样物品【经典BFS】

题目是价格范围内最高排名的 K 样物品
在这里插入图片描述
在这里插入图片描述
解题思路:

  1. 从start坐标开始遍历整张图,遍历时将当前坐标价值在合法区间的值保存下来,需要记录横、纵坐标,价格和距离四个维度因素,我们可以用一个结构体来保存
  2. 遍历完后自定义优先级排序,排序优先级分别为:距离、价格、横坐标、纵坐标,从小到大排序
  3. 如果结果数组长度小于k则全部输出,否则输出前k个

代码如下:

typedef struct node{ //自定义结构体,记录位置x,y和价格pricing和距离dist
    int x;
    int y;
    int pricing;
    int dist;
}no;
static bool cmp(no&a,no&b){ //自定义排序
    if(a.dist==b.dist){
        if(a.pricing==b.pricing){
            if(a.x==b.x){
                return a.y<b.y;
            }
            return a.x<b.x;
        }
        return a.pricing<b.pricing;
    }
    return a.dist<b.dist;
}
int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
vector<vector<int>> highestRankedKItems(vector<vector<int>>& grid, vector<int>& pricing, vector<int>& start, int k) {
    int m=grid.size();
    int n=grid[0].size();
    vector<vector<int>>ans;
    vector<vector<int>>vis(m,vector<int>(n));
    queue<pair<int,int>>q;
    vector<no>temp;
    q.push(make_pair(start[0],start[1]));
    vis[start[0]][start[1]]=1;
    int dist=0;
    while(!q.empty()){ //BFS
        int size=q.size();
        for(int i=0;i<size;i++){
            auto cur=q.front();
            q.pop();
            int x=cur.first;
            int y=cur.second;
            if(grid[x][y]>=pricing[0]&&grid[x][y]<=pricing[1]){
                temp.push_back({x,y,grid[x][y],dist});
            }
            for(int i=0;i<4;i++){
                int nx=x+dir[i][0];
                int ny=y+dir[i][1];
                if(nx>=0&&ny>=0&&nx<m&&ny<n&&!vis[nx][ny]&&grid[nx][ny]>0){
                    q.push(make_pair(nx,ny));
                    vis[nx][ny]=1;
                }
            }
        }
        dist++;
    }
    sort(temp.begin(),temp.end(),cmp);
    if(temp.size()<k){
        for(auto &node:temp){
            ans.push_back({node.x,node.y});
        }
        return ans;
    }
    for(int i=0;i<k;i++){
        ans.push_back({temp[i].x,temp[i].y});
    }
    return ans;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值