[杨小白]_leetcode_力扣第 320 场周赛-第三题

102 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第3天,点击查看活动详情

前言

小白算法比较菜,希望能激励我每日更新,从leetcode第一题开始,2022年目标1900分,1823分了!!

力扣第 320 场周赛-力扣

第 320 场周赛 - 力扣(LeetCode)

这次可真是做大牢啊,一个半小时,12分钟就做了前两题,然后干坐一个多小时,也没做出来第三题。

image.png

2477. 到达首都的最少油耗

给你一棵 n 个节点的树(一个无向、连通、无环图),每个节点表示一个城市,编号从 0 到 n - 1 ,且恰好有 n - 1 条路。0 是首都。给你一个二维整数数组 roads ,其中 roads[i] = [ai, bi] ,表示城市 ai 和 bi 之间有一条 双向路 。

每个城市里有一个代表,他们都要去首都参加一个会议。

每座城市里有一辆车。给你一个整数 seats 表示每辆车里面座位的数目。

城市里的代表可以选择乘坐所在城市的车,或者乘坐其他城市的车。相邻城市之间一辆车的油耗是一升汽油。

请你返回到达首都最少需要多少升汽油。

示例 1

image.png

输入:roads = [[0,1],[0,2],[0,3]], seats = 5

输出:3

解释:

  • 代表 1 直接到达首都,消耗 1 升汽油。
  • 代表 2 直接到达首都,消耗 1 升汽油。
  • 代表 3 直接到达首都,消耗 1 升汽油。 最少消耗 3 升汽油。

示例 2

image.png

输入:roads = [[3,1],[3,2],[1,0],[0,4],[0,5],[4,6]], seats = 2

输出:7

解释:

  • 代表 2 到达城市 3 ,消耗 1 升汽油。
  • 代表 2 和代表 3 一起到达城市 1 ,消耗 1 升汽油。
  • 代表 2 和代表 3 一起到达首都,消耗 1 升汽油。
  • 代表 1 直接到达首都,消耗 1 升汽油。
  • 代表 5 直接到达首都,消耗 1 升汽油。
  • 代表 6 到达城市 4 ,消耗 1 升汽油。
  • 代表 4 和代表 6 一起到达首都,消耗 1 升汽油。 最少消耗 7 升汽油。

提示

  • 1 <= n <= 105
  • roads.length == n - 1
  • roads[i].length == 2
  • 0 <= ai, bi < n
  • ai != bi
  • roads 表示一棵合法的树。
  • 1 <= seats <= 105

代码

这个题,略有些难想,想通了之后还是一把ac的。

image.png

  • 对于这样一张图,我们可以这样想,我们A节点的所有子节点都先集中到A节点后,一起向下一个节点运输,看需要多少辆车,就消耗多少油。

  • 对于1节点,包括他的子节点,一共有5个节点,所以把1和之后的所有节点运输到0,需要花费(5 + seats - 1) / seats(这是向上取整的写法)

  • 那么对于,3号节点,运输到1节点,同样就是(3 + seats - 1) / seats

  • 这样一顿分析后,问题就变得简单了。


  1. 第一步先建图,因为是双向图,所以两边都要建。
  2. 直接开始计算,fun函数的返回值是子节点的数目,在运输过程中会把油耗进行计算。
  3. 一把梭哈

image.png

class Solution {
    ArrayList<Integer>[] graph;
    long res = 0;

    public long minimumFuelCost(int[][] roads, int seats) {
        res = 0;
        int n = roads.length + 1; //城市的数量
        graph = buildGraph(n, roads);
        ArrayList<Integer> start = graph[0];
        for (int i = 0; i < start.size(); i++) {
            fun(0, graph[0].get(i),seats);
        }
        return res;
    }

    //计算的是,把包括nextNode和他之后的所有节点,全部运到nextNode节点后,运送到pnode需要耗费的油费
    private long fun(int pnode, Integer nextNode, int seats) {
        int nodeNum = 1; //计算nextNode之后有多少个节点,包括自己
        ArrayList<Integer> arrayList = graph[nextNode];
        for (int i = 0; i < arrayList.size(); i++) {
            if(arrayList.get(i) != pnode) {
                nodeNum += fun(nextNode, arrayList.get(i),seats);
            }
        }
        res = res + (nodeNum + seats - 1) / seats;
        return nodeNum;
    }


    private ArrayList<Integer>[] buildGraph(int n, int[][] roads) {
        ArrayList<Integer>[] graph = new ArrayList[n];
        for (int i = 0; i < graph.length; i++) {
            graph[i] = new ArrayList<Integer>();
        }
        for (int i = 0; i < roads.length; i++) {
            int a = roads[i][0];
            int b = roads[i][1];
            graph[a].add(b);
            graph[b].add(a);
        }
        return graph;
    }
}

3.结束

坐大牢,今天出分了,没掉分,1823分了。希望早日1900。knight最低是 1875.65,加油!