No.135 Candy

No.135 Candy

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:

• Each child must have at least one candy.
• Children with a higher rating get more candies than their neighbors.

What is the minimum candies you must give?

 Tags: Greedy

 

分析：分糖果问题：

每个小朋友分到的糖果数与且与他左右的两个小朋友有关【与trapping rain water相类似，但不是考虑所有所有两侧的】
当i小朋友比他旁边的rating高时，i就要比rating小的至少多一个糖果；若它比左右两侧都大，可以置为1
问题解决步骤：[辅助数组num]
1.从左往右扫描，若当前i的rating大于左侧的，i的糖果数为左侧的糖果数+1，否则，置为1
2.从右往左扫描，若当前i的rating大于右侧的，i的糖果数右侧的糖果数+1，否则，置为1
3.当前i的最小糖果数为两次扫描结果中最大的那个
4.累加所有i的糖果数
【2、3、4在同一次循环中进行】

 

 

第一个提交【错】：题意理解有问题：是相邻的，rate高的比低的糖果要多；而不是对于整个数组而言。

class Solution
{
public:
    int candy(vector<int> &ratings)
    {
        int size = ratings.size();
        if(size == 0)
            return 0;
        if(size == 1)
            return 1;
        //不知道是否允许改变原数组
        sort(ratings.begin(),ratings.end());
        
        int res=0;
        int index = 0;
        int tmp;
        int sameCount=1;//此时相同优先级的个数
        int currentCount = 1;//当前类优先级得到的candy个数
        while(index < size)
        {
            sameCount=1;
            tmp=ratings[index];
            while(index<size-1 && ratings[index+1] == tmp)//必须要保证index+1不越界!!
            {
                index++;
                sameCount++;
            }
            res += sameCount*currentCount;
            currentCount++;
            index++;//勿忘!!!
        }
        return res;
    }
};

 

最终版：

#include "stdafx.h"
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

class Solution
{
public:
    int candy(vector<int> &ratings)
    {
      /*
        分糖果问题：
        每个小朋友分到的糖果数与且与他左右的两个小朋友有关【与trapping rain water相类似，但不是考虑所有所有两侧的】
        当i小朋友比他旁边的rating高时，i就要比rating小的至少多一个糖果；若它比左右两侧都大，可以置为1
        问题解决步骤：[辅助数组num]
            1.从左往右扫描，若当前i的rating大于左侧的，i的糖果数为左侧的糖果数+1，否则，置为1
            2.从右往左扫描，若当前i的rating大于右侧的，i的糖果数右侧的糖果数+1，否则，置为1
            3.当前i的最小糖果数为两次扫描结果中最大的那个
            4.累加所有i的糖果数
            【2、3、4在同一次循环中进行】
      */

        int size = ratings.size();
        if(size == 0)
            return 0;
        if(size == 1)
            return 1;

        int *num = new int[size];
        for(int i=0; i<size; i++)
        {//从左往右扫描
            if(i>0 && ratings[i]>ratings[i-1])//注意i的范围，保证左侧不越界!!!
                num[i] = num[i-1]+1;
            else
                num[i] = 1;
        }
        int res=0;
        int cur=0;//记录对i来说，只考虑它右边的rating，所分的糖果数
        for(int i=size-1; i>=0; i--)
        {//从右往左扫描
            cur = 1;
            if(i<size-1 &&ratings[i]>ratings[i+1])
                cur = num[i+1]+1;
            res += (num[i]>cur? num[i] : cur);
            num[i] = cur;//需要保存当前的cur，以供i-1使用!!!
        }
        
        delete[] num;
        return res;
    }
};

int main()
{
    Solution sol;
//    int data[] = {1,2,5,5,10};//1+2+3+1+2=9
//    int data[] = {1,2,3,5,7};//1+2+3+4+5=15
    int data[] = {1,1,1,1,1};//5
    vector<int> test(data, data+sizeof(data)/sizeof(int));
    cout << sol.candy(test);
    return 0;
}