按照题目要求,希望left数组越小越好
根据样例可以知道,左右数组的分割点在整个数组最小值的附近
- 先找到数组最小值的的位置,以此分割出左右初数组;
- 求出左数组中的最大值max_left,右数组中的最小值min_right;
- 比较max_left和min_right的大小,找到比max_left大的min_right;
- 在找到距离整个数组最小值最近的min_right的位置;
- 得到最终的答案
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <map>
#include <set>
#include <numeric>
using namespace std;
class Solution {
public:
int partitionDisjoint(vector<int>& A)
{
vector<int>num = A;
sort(num.begin(), num.end());
if (num == A)
return 1;
num = A;
int _min = *min_element(num.begin(), num.end());
auto pos_min = find(num.rbegin(), num.rend(), _min);
int max_left = *max_element(pos_min, num.rend());
if (max_left == *pos_min)
return 1;
while (1)
{
auto temp = min_element(num.rbegin(), pos_min);
if (*temp < max_left)
{
pos_min = temp;
max_left = *max_element(pos_min, num.rend());
}
else
{
pos_min = temp;
while (*pos_min<=*temp)
{
temp++;
}
return distance(temp, num.rend());
}
}
return A.size() - 1;
}
};
int main()
{
/*int N = 100;
vector<vector<int>>edges = { {{74, 34},{67, 44},{81, 40},{1, 97},{44, 88},{95, 23},{77, 78},{67, 29},{98, 1},{89, 3},{60, 91},{30, 28},{64, 85},{47, 72},{64, 9},{26, 35},{24, 1},{43, 35},{62, 86},{92, 86},{59, 89},{31, 3},{31, 92},{1, 33},{54, 68},{57, 63},{2, 3},{36, 64},{6, 9},{3, 67},{99, 70},{9, 47},{45, 16},{94, 92},{22, 9},{56, 31},{89, 84},{40, 31},{37, 38},{57, 52},{75, 76},{1, 26},{65, 79},{5, 39},{96, 47},{55, 14},{83, 54},{6, 32},{11, 26},{8, 40},{32, 69},{32, 14},{78, 79},{34, 92},{31, 75},{39, 45},{3, 79},{71, 31},{82, 74},{51, 58},{27, 35},{60, 70},{31, 51},{53, 74},{64, 60},{84, 90},{39, 40},{28, 80},{0, 47},{31, 41},{1, 25},{56, 48},{93, 10},{1, 17},{37, 7},{47, 15},{49, 41},{5, 18},{4, 92},{25, 64},{84, 95},{10, 95},{63, 66},{46, 87},{92, 50},{66, 3},{64, 75},{61, 98},{78, 12},{54, 71},{7, 65},{87, 39},{73, 96},{61, 20},{64, 19},{21, 69},{30, 6},{42, 72},{13, 67}} };
*/
vector<int> num = { 5,0,3,4,5,4,5,5,8,6,10,7,6,10,6,7};
string S = "Test1ng-Leet=code-Q!";
Solution s;
s.partitionDisjoint(num);
//s.sumOfDistancesInTree(S);
return 0;
}
附上刚开始做的时候使用的暴力解法
class Solution {
public:
int partitionDisjoint(vector<int>& A)
{
vector<int>num = A;
auto i = num.begin() + 1;
//sort(num.begin(), num.end());
vector<int>temp1(num.begin(), i);
vector<int>temp2(i, num.end());
for (; i !=num.end(); i++)
{
int max1 = *max_element(temp1.begin(), temp1.end());
int min2 = *min_element(temp2.begin(), temp2.end());
if (max1<=min2)
{
return temp1.size();
}
}
}
};