九度OJ—题目1040：Prime Number

题目描述：

Output the k-th prime number.

输入：

k≤10000

输出：

The k-th prime number.

样例输入：

3
7

样例输出：

5
17

来源：

2008年上海交通大学计算机研究生机试真题

答疑：

解题遇到问题?分享解题心得?讨论本题请访问： http://t.jobdu.com/thread-7764-1-1.html

#include<iostream>
#include<math.h>
using namespace std;
 
inline int isprime(int x)//判断x是否为素数。此处采用内联函数形式，节省了函数调用的开销
{
    int n=(int)sqrt(x),i;
    for(i=2;i<=n;i++)
        if(x%i==0)break;  
    if(i>n)
        return 1;
    else
        return 0;
}
 
int f(int k)
{
    if(1==k) return 2;
    else
        if(2==k) return 3;
        else  
        {          
            int nextprime=f(k-1)+2;//采用递归算法，从第k-1个素数+2开始往后推算
            while (!isprime(nextprime)) nextprime+=2;//3后面的素数都为奇数，步长设置为2，减少循环次数
            return nextprime;      
        }
}
 
int main()
{
    int k;
    while(cin>>k)
    {
        cout<<f(k)<<endl;
    }
    return 0;
}
/**************************************************************
    Problem: 1040
    User: vhreal
    Language: C++
    Result: Accepted
    Time:60 ms
    Memory:1616 kb
****************************************************************/