【LeetCode - 573】松鼠模拟

1、题目描述

2、解题思路

  如果松鼠开局在树底下，那么只有唯一解，即所有坚果到树的距离乘以二（一来一回）。

  设 sum 为所有坚果到树的距离的二倍。

  则分别假设每一个坚果都是第一颗坚果，让松鼠从开局所在坐标跑到该坚果的位置，然后放到树下。

  假设 nut 为第一颗松鼠，则路程为：sum - getDistance(tree, nut) + getDistance(squirrel, nut)。

  即从树底下出发和从 squirrel 出发的差别在于去往第一颗 nut 的距离不同，其他全是一样的。

  找出最小路程即可。

3、解题代码

class Solution {
    public int minDistance(int height, int width, int[] tree, int[] squirrel, int[][] nuts) {
        int sum = 0;    // 松鼠开局在树底下的总路径和
        for (int[] nut : nuts) {
            // 一来一回，距离乘以二
            sum += 2 * getDistance(tree, nut);
        }
        // 松鼠开局在 squirrel 位置上，每个坚果分别当第一颗被采的情况
        int min = Integer.MAX_VALUE;
        int cur;
        for (int[] nut : nuts) {
            cur = sum - getDistance(tree, nut) + getDistance(squirrel, nut);
            min = Math.min(min,cur);
        }
        return min;
    }

    private int getDistance(int[] A, int[] B) {
        return Math.abs(A[0] - B[0]) + Math.abs(A[1] - B[1]);
    }
}