-
Notifications
You must be signed in to change notification settings - Fork 119
/
Copy pathKnight Probability in Chessboard.java
42 lines (33 loc) · 1.19 KB
/
Knight Probability in Chessboard.java
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
class Solution {
public double knightProbability(int n, int k, int row, int column) {
double [][]curr=new double[n][n];
double [][]next=new double[n][n];
curr[row][column]=1;
int [][]dir={{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2},{-1,-2},{-2,-1}};
for(int p=1;p<=k;p++){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(curr[i][j]!=0){
for(int d=0;d<8;d++){
int ni=i+dir[d][0];
int nj=j+dir[d][1];
if(ni<0 || nj<0 || ni>=n || nj>=n){
continue;
}
next[ni][nj]+=curr[i][j]/8.0;
}
}
}
}
curr=next;
next=new double[n][n];
}
double sum=0.0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
sum+=curr[i][j];
}
}
return sum;
}
}