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

最新推荐文章于 2020-08-25 15:56:20 发布

nupt_kwz

最新推荐文章于 2020-08-25 15:56:20 发布

阅读量398

点赞数 1

分类专栏：算法--程序员面试金典文章标签： java 编程算法程序员面试金典刷题

本文链接：https://blog.csdn.net/sinat_22797429/article/details/75577202

版权

算法--程序员面试金典专栏收录该内容

29 篇文章 0 订阅

订阅专栏

1.第K个数

题目描述

有一些数的素因子只有3、5、7，请设计一个算法，找出其中的第k个数。

给定一个数int k，请返回第k个数。保证k小于等于100。

测试样例：

返回：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;
    }
}