原题

前后两遍遍历

class Solution

{

public:

  int candy(vector<int> &ratings)

  {

    vector<int> res;

    int len = ratings.size();

    for (int i = 0; i < len; i++)

    {

      res.push_back(1);

    }

    for (int j = 0; j < len; j++)

    {

      if (j > 0 && ratings[j] > ratings[j - 1])

      {

        res[j] += res[j - 1];

      }

    }

    for (int k = len - 1; k >= 0; --k)

    {

      if (k < len - 1 && ratings[k] > ratings[k + 1] && res[k] <= res[k+1])

      {

        res[k] = res[k + 1]+1;

      }

    }

    int sum = 0;

    for (auto i : res)

    {

      cout<<i<<endl;

      sum += i;

    }

    return sum;

  }

};