o.boj 1449 第K个

注：最近这一系列ACM的内容，都是2年多之前的代码，自己回顾一下。

 

 

 

第K个   Submit: 828    Accepted:233 Time Limit: 2000MS  Memory Limit: 65536K

Description
ACMaryland这天在路上遇到了好朋友CY，好久不见，话题格外多，最近CY正在作小学奥数的家教，于是打算给ACMaryland出一道最近想到的题目，ACMaryland毕竟也是算法课的助教，加上希望在CYMM面前保持着好的印象，沉思数秒，给出了解答。如果同样是聪明的你，能在断时间内想出解决的办法吗？
正整数集合S定义如下，
1)1在集合S中；
2)如果x属于S，则2x + 1 与 3x + 1 也属于S；
3）没有其他元素属于S；
假设集合S中的元素有序排列，给定一个正整数K，你能找到第K个元素吗？



Input
一个正整数K，其中0 < k <= 10000;

Output
一个正整数K，集合S中的第K个元素。


Sample Input

2


Sample Output

3


Hint
很容易知道，前面10个元素是，1，3，4，7， 9， 10， 13， 15， 19， 21
输入2，输出第二个元素为3。


Source
Sapphire@Buptacm

 

这题自己做数据结构比较麻烦，即做一个有序的集合。。

 

但懂C++ STL的set就方便了。。

 

 

#include <iostream>
#include <set>
using namespace std;

int main()
{
    set<long> c;
    set<long>::iterator iter;
    int k, i = 1;
    
    c.insert(1);    
    iter = c.begin();
    
    while (i<20000)
    {
        c.insert(2 * (*iter) + 1);
        c.insert(3 * (*iter) + 1);
        
        iter++;
        i++;        
    }
    
    cin >> k;
    iter = c.begin();
    for (int i = 1; i < k; i++, iter++);
    
    cout << *iter << endl;
    // system("pause");
    return 0;
    
    
}