forked from lemonbashar/interviews
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPaintFence.java
More file actions
27 lines (20 loc) · 795 Bytes
/
PaintFence.java
File metadata and controls
27 lines (20 loc) · 795 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
// There is a fence with n posts, each post can be painted with one of the k colors.
// You have to paint all the posts such that no more than two adjacent fence posts have the same color.
// Return the total number of ways you can paint the fence.
// Note:
// n and k are non-negative integers.
public class PaintFence {
public int numWays(int n, int k) {
if(n <= 0) {
return 0;
}
int sameColorCounts = 0;
int differentColorCounts = k;
for(int i = 2; i <= n; i++) {
int temp = differentColorCounts;
differentColorCounts = (sameColorCounts + differentColorCounts) * (k - 1);
sameColorCounts = temp;
}
return sameColorCounts + differentColorCounts;
}
}