1018 Public Bike Management

比较复杂的模拟题,Dijstra和dfs结合,注意记牢回溯算法框架:

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define ipair pair<int,int>
int cmax,n,sp,m;
vector<vector<int>> pre;
vector<vector<ipair>> graph;
vector<int> num,dis,temp,res;
int minNeed=INT_MAX,minBack=INT_MAX;
// 每条最短路径模拟前向再反向
void dfs(int last){
    temp.push_back(last);
    if(last==0){
        int need=0,back=0;
        for(int i=temp.size()-1;i>=0;i--){
            int tid=temp[i];
            if(num[tid]>=0){
                back+=num[tid];
            }else{ //够不够补再分情况讨论
                if(back>=(-num[tid])){ //够补
                    back+=num[tid];
                }else{    //不够补,能分多少分多少
                    need+=(-num[tid]-back);
                    back=0;
                }
            }
        }
        // 根据题目这句话:Finally after another space, output the number of bikes that we must take back to PBMC after the condition of Sp is adjusted to perfect.
        if(minNeed>need){
            minNeed=need;
            minBack=back;
            res=temp;
        }else if(minNeed==need&&minBack>back){
            minBack=back;
            res=temp;
        }
    }
    for(int i=0;i<pre[last].size();i++){
        dfs(pre[last][i]);
    }
    temp.pop_back();
}

int main(){
    scanf("%d%d%d%d",&cmax,&n,&sp,&m);
    graph.resize(n+1);
    num.resize(n+1,0);
    pre.resize(n+1);
    dis.resize(n+1,INT_MAX);
    dis[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%d",&num[i]);
        num[i]-=(cmax/2);
    }
    for(int i=0;i<m;i++){
        int s,d,w;
        scanf("%d%d%d",&s,&d,&w);
        graph[s].push_back(make_pair(d,w));
        graph[d].push_back(make_pair(s,w));
    }
    priority_queue<ipair, vector<ipair>, greater<ipair>> pq;
    pq.push(make_pair(0,0)); //weight, point
    while(!pq.empty()){
        int u=pq.top().second;
        pq.pop();
        if(u==sp) break;
        for(auto& it:graph[u]){
            int v=it.first;
            int weight=it.second;
            if(dis[u]+weight<dis[v]){
                dis[v]=dis[u]+weight;
                pre[v].clear();
                pre[v].push_back(u);
                pq.push(make_pair(dis[v],v));
            }else if(dis[u]+weight==dis[v]){
                pre[v].push_back(u);
            }
        }
    }
    dfs(sp);
    printf("%d ",minNeed);
    for(int i=res.size()-1;i>=0;i--){
        printf(i==0?"%d":"%d->",res[i]);
    }
    printf(" %d",minBack);
}
相关推荐
pzx_001几秒前
【LeetCode】392.判断子序列
算法·leetcode·职场和发展
沐怡旸12 分钟前
【算法--链表】146.LRU缓存--通俗讲解
算法·面试
京东零售技术27 分钟前
查收你的技术成长礼包
后端·算法·架构
fangzelin51 小时前
算法-滑动窗口
数据结构·算法
zcz16071278212 小时前
LVS + Keepalived 高可用负载均衡集群
java·开发语言·算法
rit84324992 小时前
人工鱼群算法AFSA优化支持向量机SVM,提高故障分类精度
算法·支持向量机·分类
佩佩(@ 。 @)2 小时前
嵌入式:走馬燈-stm32GPIOF_LED9、10; GPIOE_D10、D12 流水綫蜂鸣器
stm32·嵌入式硬件·算法
知彼解己3 小时前
字符串大数相加:从初稿到优化的思路演进
java·开发语言·算法
haing20193 小时前
使用deboor法计算三次B样条曲线在参数为u处的位置的方法介绍
算法·b样条曲线·deboor