代码随想录补打卡 739 每日温度 496下一个更大元素

739 每日温度

代码如下 

func dailyTemperatures(temperatures []int) []int {   单调栈中存放的是遍历过元素的下标，并且单调栈是递增的。

          res := make([]int,len(temperatures))    //建立一个结果数组

          stack := []int{}   //设置一个单调栈

          for i,v := range temperatures {   遍历数组 

              for len(stack) != 0 && v > temperatures[stack[len(stack)-1]] {   如果这个栈不为空，且遍历到的这个元素大于栈顶元素 那么就要进行统计  temperatures[stack[len(stack)-1]] 因为stack里存放的是下标，stack[len(stack)-1]而这个就表示栈顶元素，而栈顶元素表示数组元素的下标

                  top := stack[len(stack)-1]  这里取出栈顶元素的下标  

                  stack = stack[ :len(stack)-1]   并且将这个下标出栈

                  res[top] = i - top    结果数组的下标就是当前遍历到的元素下标减去栈顶元素的下标 

              }

              stack = append(stack,i)  如果不大于栈顶元素，就把这个元素的下标放入栈中 

          }

          return res 

}

496 下一个更大元素

代码如下 

func nextGreaterElement(nums1 []int, nums2 []int) []int {

                    res := make([]int,len(nums1))  和上题一样，设置一个结果数组 

                    for i,_ := range res {    初始化结果数组，将结果数组的初始值设置为-1 

                        res[i] = -1 

                    }

                    mp := map[int]int{}  设立一个map，是将数组1中的值和对应的下标一一对应 

                    for i,v := range nums1 {  

                         mp[v] = i 

                    }

                    stack := []int{}   设置一个栈 

                    stack = append(stack,0)  因为这个栈放入的是下标，所以将nums2的第一个元素的下标放入栈中 

                    for i := 1 ; i < len(nums2) ; i++ {  遍历nums2数组 

                        for len(stack) != 0 && nums2[i] > nums2[stack[len(stack)-1]] {  如果栈不为空且当前遍历的元素大于栈顶的元素 

                               top := stack[len(stack)-1]   取出这个栈顶元素的下标 

                               if _,ok :=  mp[nums2[top]] ; ok {  因为这个是在nums2中出现的数字，需要去nums1中找是否存在，如果存在，取出对应的下标 

                                    index := mp[nums2[top]] 

                                    res[index] = nums2[i]  并将这个下标放入当前遍历过的元素 

                               }

                               stack = stack[:len(stack)-1]  出栈 

                        }

                        stack = append(stack,i)

                    }

                    return res 

}