-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathCandy.java
More file actions
47 lines (43 loc) · 912 Bytes
/
Candy.java
File metadata and controls
47 lines (43 loc) · 912 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
/**
*
*/
package cc.dectinc.leetcode;
/**
* @author chenshijiang
* @date Apr 13, 2015 4:57:59 PM
*
*/
public class Candy {
public int candy(int[] ratings) {
int length = ratings.length;
int amount = 1, last = 1, falling = 0;
for (int i = 1; i < length; i++) {
if (ratings[i] >= ratings[i - 1]) {
if (falling > 0) {
amount += falling * (falling + 1) / 2;
if (falling >= last) {
amount += falling - last + 1;
}
falling = 0;
last = 1;
}
last = ratings[i] == ratings[i - 1] ? 1 : last + 1;
amount += last;
} else {
falling++;
}
}
if (falling > 0) {
amount += falling * (falling + 1) / 2;
if (falling >= last) {
amount += falling - last + 1;
}
}
return amount;
}
public static void main(String[] args) {
Candy sol = new Candy();
int[] ratings = { 1, 2, 4, 2, 2 };
System.out.println(sol.candy(ratings));
}
}