输出第n个素数

Problem Description.png

从键盘输入得到n，输出第n个素数。
2是最小的素数（常识嘛嘻嘻）。题目描述中的k<=10000基本不需要额外的代码操作，可通过OJ。

/*
 * 输出第k个素数,比如输入3,输出5;输入7,输出17
 */

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class PrimeNumber {

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        while (scan.hasNext()) {
            // 输出第n个素数
            int n = scan.nextInt();
            int current = 2;//最小的素数是2
            int sum = 0;
            List list = new ArrayList<>();//list来保存前n个素数
            for (;;) {
                //除了2以外的所有偶数都不是素数，为了减少判断将这些偶数避开
                if (current > 2 && current % 2 == 0) {//如果current为大于2的偶数
                    current++;//current++得到奇数,利用continue进行下一次循环
                    continue;
                }
                int a = -1;
                //下面i的循环条件值得玩味,优化下得到更少的判断次数更好的性能更小的时间复杂度
                for (int i = 2; i <= Math.sqrt(current); i++) {
                    if (current % i == 0) {
                        a = 0;
                        break;
                    }
                }
                //a==-1,则当前current是素数,sum用于控制得到n个素数
                if (a == -1) {
                    sum++;
                    // System.out.println(current);
                    list.add(current);

                }
                if (sum == n) {
                    break;
                }
                current++;
            }
            //输出list的最后一个数(即第n个素数)
            System.out.println(list.get(list.size() - 1));
        }
    }
}