刷题-分糖问题

有N个孩子排队。每个孩子被分配一个评分值。
您要为这些孩子提供糖果，并遵守以下要求：
每个孩子必须至少有一颗糖果。
评分较高的儿童比邻居得到更多的糖果。

#include <iostream>
#include <vector>
using namespace std;
vector<int> ratings(10);

int candy(vector<int> &ratings) {

  //从两个方向遍历
    //先从左往右，若右边大于左边，则右边+1；
    //右边往左，若左边大于右边但是评分小于等于右边，则左边+1；
    int n=ratings.size();
    int dp[n];
    int sum=0;
    for(int i=0;i<n;i++){
        dp[i]=1;
    }
    for(int i=0;i<n-1;i++){
        if(ratings[i+1]>ratings[i]){
            dp[i+1]=dp[i]+1;
        }
    }
    for(int j=n-1;j>=0;j--){
        if((ratings[j-1]>ratings[j]&&dp[j-1]<=dp[j])){
            dp[j-1]=dp[j]+1;
        }
    }
    for(int i=0;i<n;i++){
        sum+=dp[i];
    }
    return sum;
}
int main(){
    for(int i=0;i<10;i++){
        cin>>ratings[i];
    }
    int res=candy(ratings);
    cout<<res<<endl;
}