圆环回原点问题
?
問題描述
一個環(huán)上有10個點,編號為0-9,從0點出發(fā),每步可以順時針到下一個點,也可以逆時針到上一個點,求:經(jīng)過n步又
回到0點有多少種不同的走法
舉例:
- ?
思路(動態(tài)規(guī)劃)
- 我們可以想到,再回到0點可以從右面回來,也可以從左面回來,即先到達(dá)旁邊的一個點,看看有多少回來的方法
即可。所以運(yùn)用動態(tài)規(guī)劃的思想,我們可以寫出遞推式如下:
d(k, j) = d(k-1, j-1) + d(k-1, j+1);- d(k, j)表示從點j 走k步到達(dá)原點0的方法數(shù),因此可以轉(zhuǎn)化為他相鄰的點經(jīng)過k-1步回到原點的問題,這樣將問題的規(guī)模
縮小.由于是環(huán)的問題, j-1, j+1可能會超出 0到n-1的范圍,因此,我們將遞推式改成如下:
d(k, j) = d(k-1, (j-1+n)%n) + d(k-1, (j+1)%n);- 因為問題從走k步轉(zhuǎn)化為走k-1步的問題,所以在寫程序的時候我們就按照k從0開始遞增的循環(huán)寫,這樣當(dāng)計算第k步的
時候可以直接使用k-1步的結(jié)果。
go實現(xiàn)
//n 代表點的個數(shù),k代表bushu func GetSteps(n int, k int) int {if n == 0 {return 1}if n == 2 {if n%2 == 0 {return 1} else {return 0}}var dp [100][100]intdp[0][0] = 1for i := 1; i < n; i++ {dp[0][i] = 0}for j := 1; j <= k; j++ {for i := 0; i < n; i++ {dp[j][i] = dp[j-1][(i-1+n)%n] + dp[j-1][(i+1)%n]}}return dp[k][0] }C++實現(xiàn)
/**
?* 一個圓環(huán), 有n個點, 從0出發(fā),每次只能走一步,問走k步,有多少種方法可以走回來
?*/
?
#define N 100
?
int get_step_num(int n, int k)
{
?? ?if (n==1){
?? ??? ?return 1;
?? ?}
?
?? ?if (n==2){
?? ??? ?if (k%2==0)
?? ??? ??? ?return 1;
?? ??? ?else?
?? ??? ??? ?return 0;
?? ?}
?
?? ?int arr[2][N] = {0};
?? ?int flag = 1, i = 0, j = 0;
?
?? ?arr[0][0] = 1;
?? ?for (i=1; i<n; i++) {
?? ??? ?arr[0][i] = 0;
?? ?}
?
?? ?// j is the current step
?? ?for (j=1; j<=k; j++) {
?? ??? ?for (i=0; i<n; i++) {
?? ??? ??? ?arr[flag][i] = arr[!flag][(i-1+n)%n] + arr[!flag][(i+1)%n];
?? ??? ?}
?? ??? ?flag = !flag;
?? ?}
?? ?
?? ?return arr[!flag][0];
}
?
int main()
{
?? ?int n, k;
?? ?printf("Please input the number n and the step k:\n");
?? ?scanf("%d%d", &n, &k);
?
?? ?printf("%d\n", get_step_num(n, k));
?
?? ?return 0;
}
?
總結(jié)
- 上一篇: 数字n,按字典排序,找出第k小的数字
- 下一篇: 二叉树的遍历 (递归和非递归实现)