题目
一贫如洗的椎夫阿里巴巴在去砍柴的路上,无意中发现了强盗集团的藏宝地,藏宝地有编号从0-N的子,每个箱子上面有一个数字,箱子排列成一个环,编号最大的箱子的下一个是编号为0的箱子。请输出每个箱子贴的数字之后的第一个比它大的数,如果不存在则输出-1。
输入描述
输入一个数字字串,数字之间使用逗号分隔,例如: 1,2,3,1
1≤字串中数字个数≤10000:
-100000≤每个数字值≤100000
输出描述
下一个大的数列表,以逗号分隔,例如: 2,3,6,-1,6
示例1:
输入
2,5,2
输出
5,-1,5
说明
第一个2的下一个更大的数是5。数字5找不到下一个更大的数。第二个2的下一个最大的数需要循环搜索,结果也是5。
示例2:
输入
3,4,5,6,3
输出
4,5,6-1.4
思路
方法1:最直观的暴力法
题目要求找比当前值更大的下个值,那么可以搞两层循环
外层循环i的范围为[0,nums.length-1],i代表在nums中的索引,nums代表输入的列表
对于nums[i]来说,要找下一个比他更大的值,找一圈(j不能等于自身,j!=i,考虑循环查找,可以用求余实现:j=(j+1)%nums.length)后如果还找不到比当前值(nums[i])更大的值,那么返回-1,结束查找。所以内存循环条件可以写为:while (j != i && nums[j] <= nums[i]) {j = (j + 1) % nums.length;}
内存while循环结束后:
- 如果j等于i,说明找了一圈还是没找到,那么res[i]=-1,
- 如果j!=i,那么res[i]=nums[j],
最后将res使用逗号连接返回即可
方法二:单调栈法
凡是求下一个更大或者更小的题,都可以考虑单调栈。
leetcode原题:503. 下一个更大元素 II
题解
package hwod;
import java.util.*;
public class FindGoldBox4 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
String[] inputs = sc.nextLine().split(",");
int[] boxs = Arrays.stream(inputs).mapToInt(Integer::parseInt).toArray();
System.out.println(findGoldBox2(boxs));
}
//暴力法
private static String findGoldBox(int[] boxs) {
int size = boxs.length;
int[] res = new int[size];
for (int i = 0; i < size; i++) {
int j = (i + 1) % size;
while (j != i && boxs[j] <= boxs[i]) {
j = (j + 1) % size;
}
if (j == i) res[i] = -1;
else res[i] = boxs[j];
}
StringBuilder sb = new StringBuilder();
for (int i = 0; i < size; i++) {
if (i != 0) sb.append(",");
sb.append(res[i]);
}
return sb.toString();
}
//单调栈法
public static String findGoldBox2(int[] nums) {
int size = nums.length;
int[] res = new int[size];
Arrays.fill(res, -1);
Deque<Integer> stack = new LinkedList<>();
for (int i = 0; i < size * 2 - 1; i++) {
while (!stack.isEmpty() && nums[stack.peek()] < nums[i % size]) {
res[stack.pop()] = nums[i % size];
}
if (i < size) stack.push(i % size);//相比leetcode,多加一个判断,避免重复出入栈
}
StringBuilder sb = new StringBuilder();
for (int i = 0; i < size; i++) {
if (i != 0) sb.append(",");
sb.append(res[i]);
}
return sb.toString();
}
}
推荐
如果你对本系列的其他题目感兴趣,可以参考华为OD机试真题及题解(JAVA),查看当前专栏更新的所有题目。