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 |
반응형