PAT 乙级 1005

题目

    题目地址：PAT 乙级 1005

 

题解

    本题主要就在于将一个数的一系列计算结果不重复地存储起来并便于检索，考虑到STL中的集合有相似的特性，使用set可以有效地简化代码和运算。

    过程如下：

      （初始化两个集合，结果集合result和临时数据集合stmp）

        1. 取第一个元素，判断其是否存在于result集合，不存在则将其加入result和stmp集合；

        2. 进行运算，运算结果存入stmp集合；

        3. 每一次运算结束后，判断运算结果是否存在于result集合中，如果存在则从result集合中删除该数；

        4. 当运算结果为1时，结束该元素的运算，取下一个元素，重复1、2、3步骤。

 

代码

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

int main() {
    set<int> result;
    set<int> stmp;
    int k = 0;
    int num[105];
    cin >> k;
    for (int i = 0; i < k; i++)
        cin >> num[i];
    for (int i = 0; i < k; i++) {
        int tmp = 0;
        if (!stmp.count(num[i])) {
            result.insert(num[i]);
            stmp.insert(num[i]);
        }
        while (num[i] != 1) {
            if (num[i] % 2 == 0)
                num[i] /= 2;
            else
                num[i] = (num[i] * 3 + 1) / 2;
            stmp.insert(num[i]);
            if (result.count(num[i])) {
                set<int>::iterator it;
                it = result.find(num[i]);
                result.erase(it);
            }
        }
    }
    int size = result.size() - 1;
    set<int>::reverse_iterator rit;
    for (rit = result.rbegin(); rit != result.rend(); rit++) {
        cout << *rit;
        if (size--)
            cout << ' ';
    }
    cout << endl;

    return 0;
}