程序员面试金典:第K个数、机器人走方格

1.第K个数

3

返回：7

方法1：//用动态规划来求解：（参考牛客网牛油解题）

public class Main {
public static void main(String[] args) {
System.out.println(findKth(8));
}

public static int findKth(int k) {
int[] re=new int[k];
if(k==1) return 3;
if(k==2) return 5;
if(k==3) return 7;
re[0]=3;
re[1]=5;
re[2]=7;
//用num3,num5,num7来做标记
int num3=0;
int num5=0;
int num7=0;
for(int i=3;i<k;i++){
re[i]=Math.min(re[num3]*3, Math.min(re[num5]*5, re[num7]*7));
if(re[i]==re[num3]*3) num3++;
if(re[i]==re[num5]*5) num5++;
if(re[i]==re[num7]*7) num7++;
}
return re[k-1];
}
}

方法2：小于当前值将当前指针往后移

import java.util.*;

public class KthNumber {
public int findKth(int k) {
// write code here
if (k < 0)
return -1;
int[] res = new int[k+1];
res[0] = 1;
int p3 = 0;
int p5 = 0;
int p7 = 0;
for (int i=1; i <= k; i++) {
res[i] = Math.min(res[p3] * 3, Math.min(res[p5] * 5, res[p7] * 7));
while (res[p3] * 3 <= res[i])
p3++;
while (res[p5] * 5 <= res[i])
p5++;
while (res[p7] * 7 <= res[i])
p7++;
}
return res[k];
}
}

方法3：通过队列来实现

import java.util.LinkedList;
import java.util.Queue;

public class Main2 {
public static void main(String[] args) {
System.out.println(findKth(8));
}

public static int findKth(int k) {
int val=0;
Queue<Integer> q3=new LinkedList<Integer>();
Queue<Integer> q5=new LinkedList<Integer>();
Queue<Integer> q7=new LinkedList<Integer>();
q3.add(3);
q5.add(5);
q7.add(7);
for(int i=0;i<k;i++){
if(!q3.isEmpty()&&!q5.isEmpty()&&!q7.isEmpty()){
int v3=q3.peek(),v5=q5.peek(),v7=q7.peek();
val=Math.min(v3,Math.min(v5,v7));
if(val==v3){
q3.remove();
q3.add(val*3);
q5.add(val*5);
q7.add(val*7);
}
if(val==v5){
q5.remove();
q5.add(val*5);
q7.add(val*7);
}
if(val==v7){
q7.remove();
q7.add(val*7);
}
}
}
return val;
}
}

2.机器人走方格

2,2

返回：2