[LeetCode]624. Maximum Distance in Arrays

[LeetCode]624. Maximum Distance in Arrays

题目描述

思路

保存当前最大最小和第二大第二小的值

如果最大最小值在同一个数组里就返回max(最大 - 第二小， 第二大 - 最小)

更新当前最大最小和第二大第二小的过程

当更新最大最小值时，先将目前最大最小值保存到第二大第二小， 再更新

当无法更新最大最小值时，检查第二大第二小能否符合更新条件，更新条件除去更大更小之外也包括不能与当前最大最小相同

以上更新均更新下标，即第几个数组中获取到的最大最小值

代码

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
int maxDistance(vector<vector<int>>& arrays) {
int curMin = 0, nextMin = 1, curMax = 0, nextMax = 1;
for (int i = 0; i < arrays.size(); ++i) {
if (arrays[i].front() < arrays[curMin].front()) {
nextMin = curMin;
curMin = i;
}
else if (arrays[i].front() <= arrays[nextMin].front() && i != curMin) {
nextMin = i;
}

if (arrays[i].back() > arrays[curMax].back()) {
nextMax = curMax;
curMax = i;
}
else if (arrays[i].back() >= arrays[nextMax].back() && i != curMax) {
nextMax = i;
}
}

return curMin == curMax ? max(abs(arrays[nextMax].back() - arrays[curMin].front()), abs(arrays[curMax].back() - arrays[nextMin].front())) : abs(arrays[curMax].back() - arrays[curMin].front());
}
};

int main() {
vector<vector<int>> arrays = { {-1, 1},{-3, 4}, {-2, 2} };
Solution s;

cout << s.maxDistance(arrays) << endl;

system("pause");
return 0;
}