135.分发糖果

题目

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

C++

#include <iostream>
#include <cstring>
#include <cmath>
#include <vector>
#include <list>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <mutex>
#include <thread>
#include <algorithm>
using namespace std;


int main(){
	printf("n:");
	int n;
	scanf("%d",&n);
	vector<int> ratings(n);
	vector<int> candies(n,1);

	for( int i=0;i<n;i++ ){
		printf("ratings%d:",i+1);
		int temp1;
		scanf("%d",&temp1);
		ratings[i]=temp1;
	}

	int size=ratings.size();
	int count=0;
	for( int i=1;i<size;i++ ){
		if(ratings[i]>ratings[i-1]){
			candies[i]=candies[i-1]+1;
		}
	}
	for( int i=size-1;i>0;i-- ){
		if( ratings[i-1]>ratings[i] ){
			candies[i-1]=max(candies[i-1],candies[i]+1);
		}
	}

	for( int i=0;i<size;i++ ){
		count=count+candies[i];
	}

	printf("%d.",count);
	return 0;
}

时间复杂度

$O(N)$

空间复杂度

$O(N)$

Java

import java.util.Scanner;
import java.lang.Math;
import java.util.Vector;

public class java135{

	

	public static void main(String [] args){
		Scanner s=new Scanner(System.in);
		System.out.println("n:");
		int n=s.nextInt();
		Vector<Integer> ratings=new Vector<Integer>(n);
		Vector<Integer> candies=new Vector<Integer>(n);
		for( int i=0;i<n;i++  ){
			candies.add(1);
			System.out.println("ratings"+(i+1)+":");
			int temp=s.nextInt();
			ratings.add(temp);
		}
		int size=ratings.size();
		int count=0;
		for( int i=1;i<size;i++ ){
			if( ratings.get(i)>ratings.get(i-1) ){
				candies.set(i,candies.get(i-1)+1);
			}
		}
		for( int i=size-1;i>0;i-- ){
			if( ratings.get(i-1)>ratings.get(i) ){
				candies.set(i-1,Math.max(candies.get(i-1),candies.get(i)+1));
			}
		}
		for( int i=0;i<size;i++ ){
			count=count+candies.get(i);
		}
		System.out.println(count+".");
	}
}

时间复杂度

$O(N)$

空间复杂度

$O(N)$

Python

import math;
print("n:");
n=int(input());
ratings=[0 for _ in range(n)];
candies=[1 for _ in range(n)];

for i in range(n):
	print("ratings"+str(i+1)+":");
	ratings[i]=int(input());

count=0;
size=len(ratings);
for i in range(1,size):
	if ratings[i]>ratings[i-1]:
		candies[i]=candies[i-1]+1;
for i in range(size-1,0,-1):
	if ratings[i-1]>ratings[i]:
		candies[i-1]=max(candies[i-1],candies[i]+1);
for i in range(size):
	count=count+candies[i];
print(str(count)+".\n");

时间复杂度

$O(N)$

空间复杂度

$O(N)$