carloscn / structstudy

Leetcode daily trainning by using C/C++/RUST programming.
4 stars 1 forks source link

leetcode1184:公交站间的距离(distance_between_bus_stops) #192

Open carloscn opened 1 year ago

carloscn commented 1 year ago

问题描述

环形公交路线上有 n 个站,按次序从 0 到 n - 1 进行编号。我们已知每一对相邻公交站之间的距离,distance[i] 表示编号为 i 的车站和编号为 (i + 1) % n 的车站之间的距离。

环线上的公交车都可以按顺时针和逆时针的方向行驶。

返回乘客从出发点 start 到目的地 destination 之间的最短距离。

示例 1:

image

输入:distance = [1,2,3,4], start = 0, destination = 1 输出:1 解释:公交站 0 和 1 之间的距离是 1 或 9,最小值是 1。  

示例 2:

image

输入:distance = [1,2,3,4], start = 0, destination = 2 输出:3 解释:公交站 0 和 2 之间的距离是 3 或 7,最小值是 3。  

示例 3:

image

输入:distance = [1,2,3,4], start = 0, destination = 3 输出:4 解释:公交站 0 和 3 之间的距离是 6 或 4,最小值是 4。  

提示:

1 <= n <= 10^4 distance.length == n 0 <= start, destination < n 0 <= distance[i] <= 10^4

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/distance-between-bus-stops 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/distance-between-bus-stops

carloscn commented 1 year ago

问题分析

这个用循环链表来做比较好一点,但是没必要。我们直接用数组做,在一个循环中,分为同向和逆向两个方向,分别达到自己的条件才循环下一个,两个向用count计数,最后比较计数求最小值即可。

pub fn distance_between_bus_stops(distance: Vec<i32>, start: i32, destination: i32) -> i32
{
    let mut ret = 0;

    if distance.len() <= destination as usize ||
       start > destination {
        return -1;
    }

    if distance.len() < 1 || start == destination {
        return ret;
    }

    let mut pos_count:i32 = 0;
    let mut neg_count:i32 = 0;
    for i in 0..distance.len() as i32 {
        if i >= start && i < destination {
            pos_count += distance[i as usize];
        } else {
            neg_count += distance[i as usize];
        }
    }

    ret = pos_count.min(neg_count);

    return ret;
}
carloscn commented 1 year ago

code

https://review.gerrithub.io/c/carloscn/structstudy/+/552857 https://github.com/carloscn/structstudy/commit/6881178f36decd3299c063098ed2a226c5d263bf