135. Problema de distribución de caramelos
Descripción del problema
La maestra quiere distribuir dulces a los niños, hay N niños en línea recta y la maestra calificará a cada niño por adelantado según su desempeño.
Debe ayudar al maestro a distribuir dulces a estos niños de acuerdo con los siguientes requisitos:
每个孩子至少分配到 1 个糖果。
评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
Entonces, ¿cuántos dulces necesita preparar el maestro al menos?
Ejemplo 1:
输入:[1,0,2]
输出:5
解释:你可以分别给这三个孩子分发 2、1、2 颗糖果。
Ejemplo 2:
输入:[1,2,2]
输出:4
解释:你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
respuesta
Pensamiento codicioso: Primero, cada niño tiene un dulce por persona
. La operación de clasificación no se puede realizar considerando el número de dulces asignados y la posición.
- Subproblemas de segmentación: considere solo una dirección a la vez
- Desplácese de izquierda a derecha. Si la puntuación de la derecha es alta, asigne el número de caramelos num [i] al número de caramelos del niño de la izquierda + 1 num [i-1] +1
- Recorrido inverso de derecha a izquierda Cuando la izquierda es mayor que la derecha, compare en este momento max (num [i-1], num [i] +1) Actualizar num [i-1]
Código
class Solution {
public int candy(int[] ratings) {
// 问题定位:贪心
// 和身边的人有关 不能改变顺序
// 最优解:找到老师需要准备的最小糖果数
// 子问题:两次遍历 先左右,后右左 每次找到应分配的糖果数
int N = ratings.length;
int[] num =new int[N];
int ret=0;
// 1.每个孩子先都给一个糖
for(int i=0; i<N ; i++){
num[i] = 1 ;
}
//2.左到右遍历
for(int i=1;i<N;i++){
if(ratings[i] > ratings[i-1]){
num[i] = num[i-1] + 1 ;
}
}
//3.从右向左
for(int i=N-1;i>0;i--){
if(ratings[i] < ratings[i-1]){
num[i-1] =Math.max(num[i-1],num[i]+1);
}
}
// 4.计算数目
for(int i=0;i<N; i++){
ret += num[i];
}
return ret;
}
}