본문 바로가기

알고리즘/BOJ

1551번

1551번 - 수열의 변화


조건에 수열의 숫자의 크기에 대한 조건이 없기도 하고, 그래서 그냥 string 을 써서 문제를 풀었다.


조금 더럽게 푼 감이 있긴 하지만,


stoi 함수를 이용해서 입력받은 string을 int 형 변수로 바꿔줬다. (int 형으로 풀리는거 보니 엄청 큰 수는 없는 듯 함)


바꾼 int 값들을 v 벡터에 모두 넣어주고, 주어진 조건을 이용해서 v[i] = v[i+1] - v[i] 값을 다시 저장했다.


그리고 맨 끝 값은 필요없기 때문에 v.pop_back() 을 통해서 마지막값을 없애줬다.


<정답 코드>


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
#include<iostream>
#include<string>
#include<vector>
 
using namespace std;
 
int main()
{
    int n, k;
    vector<int> v;
    vector<int> ans;
    string s, tmp = "";
    cin >> n >> k;
    cin >> s;
    
    for (int i = 0; i < s.size(); i++)
    {    
        if ((s[i] >= '0' && s[i] <= '9')||s[i]=='-')
        {
            tmp += s[i];
        }
        if (s[i] == ',' || i==s.size()-1)
        {
            v.push_back(stoi(tmp));
            tmp = "";
        }
    }
 
    for (int i = 0; i < k; i++)
    {
        for (int j = 0; j < v.size()-1; j++)
        {
            v[j] = v[j + 1- v[j];
        }
        v.pop_back();
    }
 
    for (int i = 0; i < v.size(); i++)
    {
        cout << v[i];
 
        if (i == v.size() - 1)
        {
            break;
        }
        cout << ",";
    }
    return 0;
}
cs


반응형

'알고리즘 > BOJ' 카테고리의 다른 글

9517번  (0) 2018.02.27
1526번  (0) 2018.02.27
5525번  (0) 2018.02.27
1913번  (0) 2018.02.26
10828번  (0) 2018.02.25