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);
}
相关推荐
Kidddddult3 分钟前
力扣刷题Day 43:矩阵置零(73)
算法·leetcode·力扣
大龄Python青年2 小时前
C语言 交换算法之加减法,及溢出防范
c语言·开发语言·算法
啊我不会诶2 小时前
CF每日5题
算法
朱剑君3 小时前
排序算法——基数排序
算法·排序算法
COOCC13 小时前
PyTorch 实战:从 0 开始搭建 Transformer
人工智能·pytorch·python·深度学习·算法·机器学习·transformer
拾忆-eleven4 小时前
C++算法(19):整数类型极值,从INT_MIN原理到跨平台开发实战
数据结构·c++·算法
JK0x075 小时前
代码随想录算法训练营 Day39 动态规划Ⅶ 打家劫舍
算法·动态规划
blammmp5 小时前
算法专题四:前缀和
java·开发语言·算法
望未来无悔6 小时前
系统学习算法:动态规划(斐波那契+路径问题)
java·算法
明月看潮生6 小时前
青少年编程与数学 02-018 C++数据结构与算法 25课题、图像处理算法
c++·图像处理·算法·青少年编程·编程与数学