496.下一个更大元素 I

题目：
nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x ****大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 **ans **作为答案，满足 **ans[i] **是如上所述的 下一个更大元素 。
解法：
涉及比较数组当前元素和下一个元素时，可以考虑转换为比较前一个和当前元素，代码会不会更好写

func nextGreaterElement(nums1 []int, nums2 []int) []int {
	ans := make([]int, len(nums1))
	nextGreaterM := make(map[int]int)
	stack := make([]int, len(nums2))
	for i := range nums2 {
		for len(stack) != 0 && nums2[stack[len(stack) - 1]] < nums2[i] {
			nextGreaterM[nums2[stack[len(stack) - 1]]] = nums2[i]
			stack = stack[:len(stack) - 1]
		}
		stack = append(stack, i)
	}
	for i := range nums1 {	
		if nextGreaterVal, ok := nextGreaterM[nums1[i]]; ok {
			ans[i] = nextGreaterVal
		} else {
			ans[i] = -1
		}
	}
	return ans
}