松鼠模拟
题目描述
现在有一棵树,一只松鼠和一些坚果。位置由二维网格的单元格表示。你的目标是找到松鼠收集所有坚果的最小路程,且坚果是一颗接一颗地被放在树下。松鼠一次最多只能携带一颗坚果,松鼠可以向上,向下,向左和向右四个方向移动到相邻的单元格。移动次数表示路程。
输入 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<<sum;
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;
}
};