leetcode135分发糖果

leetcode135 Candy

Posted by BY on May 30, 2018

前言

暴力可以解,但是优化非常需要技巧

正文

问题来源

本问题来自leetcode上的135题。

问题描述

老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:

  • 每个孩子至少分配到 1 个糖果。
  • 相邻的孩子中,评分高的孩子必须获得更多的糖果。
    那么这样下来,老师至少需要准备多少颗糖果呢?

    示例 1:

    输入: [1,0,2]
    输出: 5
    解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
    

    示例 2:

    输入: [1,2,2]
    输出: 4
    解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
    

    分析:

    假设每个孩子分到的糖果数组为A[N],初始化为{1},因为每个人至少分到一颗糖。
    1、与前面的邻居比较,前向遍历权重数组ratings,如果ratings[i]>ratings[i-1],则A[i]=A[i-1]+1;
    2、与后面的邻居比较,后向遍历权重数组ratings,如果ratings[i]>ratings[i+1]且A[i]<A[i+1]+1,则更新A,A[i]=A[i+1]+1;
    3、对A求和即为最少需要的糖果。
    时间复杂度:O(n)
    空间复杂度:O(n)
    C代码如下:

    int candy(int* ratings, int ratingsSize) {
      int *dp = NULL;
      int i;
      int sum=0;
      dp = (int*)malloc(sizeof(int)*ratingsSize);
      for(i=0;i<ratingsSize;i++)//初始化
      {
          dp[i] = 1;
      }
      for(i=1;i<ratingsSize;i++)//正向
      {
          if(ratings[i-1]<ratings[i])
          {
              dp[i] = dp[i-1] + 1;
          }
      }
      for(i=ratingsSize-2;i>=0;i--)//反向
      {
          if(ratings[i+1]<ratings[i])
          {
              dp[i] = max(dp[i],dp[i+1] + 1);
          }
      }
      for(i=0;i<ratingsSize;i++)//求和
      {
          sum += dp[i];
      }
      free(dp);
      return sum;
    }
    

    总结:

    凭感觉做的。网上把这题归为贪心。

    结语

    闲来无事的话,多练习练习,做做题。