(LeetCode 135) Candy

6311 단어 LeetCode
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?
     

    제목:


    N명의 아이들이 한 줄로 서서 모든 사람에게 권중을 설정한다.다음과 같은 규칙에 따라 사탕을 분배한다. (1) 모든 아이가 적어도 사탕 한 알을 나누면 (2) 권중이 비교적 높은 아이는 그의 이웃보다 더 많은 사탕을 얻을 수 있다.
    물음: 모두 최소 몇 개의 사탕이 필요합니까?알고리즘의 사고방식과 알고리즘의 시간, 공간의 복잡도가 얼마나 되는지 분석하세요.
     

    생각:


    가령 모든 아이가 나눈 사탕 수조가 A[N]이고 초기화는 {1}이다. 왜냐하면 모든 사람이 적어도 사탕 한 알을 나누기 때문이다.
    방법 1:
    1. 앞의 이웃과 비교하면 전방향 역류권 중수조ratings, 만약ratings[i]>ratings[i-1], A[i]=A[i-1]+1;
    2. 뒤에 있는 이웃과 비교하면 후방 역류권 중수조ratings, 만약ratings[i]>ratings[i+1]와 A[i]3. A에게 가장 필요한 사탕을 구한다.
    시간 복잡도: O(n)
    공간 복잡도: O(n)
    방법 2:
    수조 A1을 설정하여 전방향적인 결과를 저장하고 수조 A2를 설정하여 후향적인 결과를 저장한다.
    1. 앞의 이웃과 비교하면 전방향 역류권 중수조ratings, 만약ratings[i]>ratings[i-1], A1[i]=A1[i-1]+1;
    2. 뒤에 있는 이웃과 비교하면 후방 역류권 중수조ratings, 만약ratings[i]>ratings[i+1], A2[i]=A2[i+1]+1;
    3. 모든 아이가 앞뒤 이웃보다 사탕을 많이 나누는데 A1, A2의 비교적 큰 사람, 즉 A[i]=max(A1[i], A2[i])이다.
    4. A에게 가장 필요한 사탕을 구한다.
    시간 복잡도: O(n)
    공간 복잡도: O(n)

    코드:

    class Solution {
    public:
        int candy(vector<int>& ratings) {
            int n=ratings.size();
            int sum;
            
            vector<int> nums(n,1);
            for(int i=1;i<n;i++){
                if(ratings[i]>ratings[i-1])
                    nums[i]=nums[i-1]+1;
            }
            
            sum=nums[n-1];
            for(int i=n-2;i>=0;i--){
                if(ratings[i]>ratings[i+1] && nums[i]<nums[i+1]+1)
                    nums[i]=nums[i+1]+1;
                sum+=nums[i];
            }
            
            return sum;
        }
    };
    class Solution {
    public:
        int candy(vector<int>& ratings) {
            int n=ratings.size();
            int sum=0;
            
            vector<int> nums(n,1);
            vector<int> for_nums(n,1);
            vector<int> back_nums(n,1);
            
            for(int i=1;i<n;i++){
                if(ratings[i]>ratings[i-1])
                    for_nums[i]=for_nums[i-1]+1;
            }
            
            for(int i=n-2;i>=0;i--){
                if(ratings[i]>ratings[i+1])
                    back_nums[i]=back_nums[i+1]+1;
            }
            
            for(int i=0;i<n;i++){
                nums[i]=max(for_nums[i],back_nums[i]);
                sum+=nums[i];
            }
            
            return sum;
        }
    };

    좋은 웹페이지 즐겨찾기