알고리즘/알고리즘 문제

[삼성 swea] 보물상자 비밀번호

charminseok 2019. 9. 22. 00:41

맨 뒤 글자가 맨 앞으로 이동하면서 키를 만든다. 여기서 n/4개씩 끊어서 set에 넣어줬다.(중복을 없애기 위해서)

그리고 set은 오름차순으로 들어가게 set<string,greater<string>> strtmp;

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#include<iostream>
#include<set>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
 
int n, k;
set<string,greater<string>> strtmp;
vector<string> ans;
 
void rotate(string &str) {
    string tmp = "";
    tmp += str[n - 1];
    str = str.substr(0,n - 1);
    str = tmp.append(str);
    for (int i = 0; i < n; i = i + (n/4)) {
        strtmp.insert(tmp.substr(i, n / 4));
    }
}
long long hex(string str) {
    long long ret = 0;
    int h = 1;
    for (int i = str.size() - 1; i >= 0; i--) {
        char tmp = str[i];
        if (tmp >= '0' && tmp <= '9') ret += (tmp - '0')*h;
        else if (tmp >= 'A' && tmp <= 'F') ret += (tmp - 55)*h;
        h *= 16;
    }
    return ret;
}
int main(int argc, char** argv)
{
    int test_case;
    int T;
    string str;
    freopen("sample_input.txt""r", stdin);
    cin >> T;
    for (test_case = 1; test_case <= T; ++test_case)
    {
        cin >> n >> k;
        cin >> str;
        for(int i =0; i <(n/4);i++)
            rotate(str);
 
        set<string>::iterator iter;
        for (iter = strtmp.begin(); iter != strtmp.end(); iter++) {
            ans.push_back(*iter);
        }
 
        long long answer = hex(ans[k-1]);
 
        cout << "#" << test_case << " " << answer << endl;
        strtmp.clear();
        ans.clear();
    } 
    return 0;//정상종료시 반드시 0을 리턴해야합니다.
}
http://colorscripter.com/info#e" target="_blank" style="color:#4f4f4ftext-decoration:none">Colored by Color Scripter