分糖果问题

import java.util.*;
public class Solution {
  /**
   * pick candy
   * @param arr int整型一维数组 the array
   * @return int整型
   */
  public int candy (int[] arr) {
      // write code here
      int[] tmp= new int[arr.length];
      Arrays.fill(tmp,1);
      int count=0;

      for(int i=1;i<arr.length;i++){
          if(arr[i]>arr[i-1]){
              tmp[i]=tmp[i-1]+1;
          }
      }
       for(int i=arr.length-1;i>0;i--){
          if(arr[i-1]>arr[i]){
              tmp[i-1]=Math.max(tmp[i-1],tmp[i]+1);
          }
      }
      for(int i:tmp)
          count+=i;
      return count;
  }
}