/*
由范围 [0,n] 内所有整数组成的 n + 1 个整数的排列序列可以表示为长度为 n 的字符串 s ,其中:
如果 perm[i] < perm[i + 1] ,那么 s[i] == 'I'
如果 perm[i] > perm[i + 1] ,那么 s[i] == 'D'
给定一个字符串 s ,重构排列 perm 并返回它
输入:s = "IDID"
输出:[0,4,1,3,2]
*/
//简单贪心算法,每次遇到'I',就取0-n中的最小值,
//遇到'D'就取0-n中的最大值
public:
vector<int> diStringMatch(string s) {
int n = s.length(), lo = 0, hi = n;
int[] perm = new int[n+1];
for(int i = 0; i < n; i++){
perm[i] = s.charAt(i) == 'I' ? lo++:hi++;
}
perm[n] = lo;//此时lo和hi相等,都为剩下的那个数
return perm;
}
/*
给你一个整数数组 nums ,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1
输入:nums = [1, 7, 3, 6, 5, 6]
输出:3
*/
//求前缀和,记数组的全部元素之和为{total}
//当遍历到第 i 个元素时,设其左侧元素之和为 {sum}
//判断sum*2+num[i]==total
public int pivotIndex(int[] nums) {
int total = 0;
for(int i = 0; i < nums.length; ++i){
total += nums[i];
}
int sum = 0;
for(int i = 0; i < nums.length; i++){
if(sum * 2 + nums[i] == total){
return i;
}
sum += nums[i];
}
return -1;
}
};