leetcode菜狗入门 | 573. 松鼠模拟

松鼠模拟

题目描述

现在有一棵树，一只松鼠和一些坚果。位置由二维网格的单元格表示。你的目标是找到松鼠收集所有坚果的最小路程，且坚果是一颗接一颗地被放在树下。松鼠一次最多只能携带一颗坚果，松鼠可以向上，向下，向左和向右四个方向移动到相邻的单元格。移动次数表示路程。

输入 1:

输入: 
高度 : 5
宽度 : 7
树的位置 : [2,2]
松鼠 : [4,4]
坚果 : [[3,0], [2,5]]
输出: 12

​​​​​
注意:

所有给定的位置不会重叠。
松鼠一次最多只能携带一颗坚果。
给定的坚果位置没有顺序。
高度和宽度是正整数。 3 <= 高度 * 宽度 <= 10,000。
给定的网格至少包含一颗坚果，唯一的一棵树和一只松鼠。

解题思路

总路程 = 所有松果到松树的距离 * 2 - 某个松果到松树的距离 + 该松果到松鼠的距离，选择最小的返回即可

代码

class Solution {
public:
    int minDistance(int height, int width, vector<int>& tree, vector<int>& squirrel, vector<vector<int>>& nuts) {
        vector<int> songshu;
        vector<int> shu;
        int ans = 0x7fffffff;
        int sum = 0;
        for(int i = 0 ; i < nuts.size() ; ++ i){
            int d1 = distance(nuts[i],squirrel); //松鼠到松果的距离
            songshu.push_back(d1);
            int d2 = distance(nuts[i], tree); //松果到树的距离
            shu.push_back(d2);
            sum += 2 * d2;
        }
        //cout<
        int index = 0;
        for(int i = 0 ; i < nuts.size() ; ++ i){
            if(ans > sum-shu[i]+songshu[i]){
                ans = sum-shu[i]+songshu[i];
                index = i;
            }
           // ans = min (ans, sum-shu[i]+songshu[i]);
        }
        cout<<index;
        return ans;
    }
    int distance(vector<int> a, vector<int> b){
        int x = abs(a[0] - b[0]);
        int y = abs(a[1] - b[1]);
        int ans = x + y;
        return ans;
    }
};