leetcode - 135. 分发糖果

最新推荐文章于 2024-07-16 23:15:32 发布

zmm_mohua

最新推荐文章于 2024-07-16 23:15:32 发布

阅读量93

点赞数

分类专栏：刷题文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/qq_42250642/article/details/125255897

版权

刷题专栏收录该内容

312 篇文章 0 订阅

订阅专栏

leetcode - 135. 分发糖果

题目

在这里插入图片描述

代码

#include <iostream>
#include <vector>
using namespace std;

// 1：ratings[i] > ratings[i-1]   从左往右 
// 2: ratings[i] > ratings[i+1]   从右往左 
int candy(vector<int>& ratings) {
	int n = ratings.size();
	if(n == 1){
		return 1;
	}
	int pre, cur;
	int count = 0;
	vector<int> left(n);
	if(ratings[0] > ratings[1]){
		pre = 2;
	}else{
		pre = 1;
	}
	left[0] = pre; 
	for(int i = 1; i < n; i++){
		if(ratings[i] > ratings[i-1]){
			cur = pre + 1;
		}else{
			cur = 1;
		}
		pre = cur;
		left[i] = pre;
	} 
	if(ratings[n-1] > ratings[n-2]){
		pre = 2;
	}else{
		pre = 1;
	}
	count += max(pre, left[n-1]);
	for(int i = n - 2; i >= 0; i--){
		if(ratings[i] > ratings[i+1]){
			cur = pre + 1;
		}else{
			cur = 1;
		}
		pre = cur;
		count += max(pre, left[i]);
	}
	return count;
}

int main(){
	int n, res;
	cin>>n;
	vector<int> ratings(n);
	for(int i = 0; i < n; i++){
		cin>>ratings[i];
	}
	res = candy(ratings);
	cout<<res;
	return 0;
}