백준 1551번: 수열의 변화 (Python, C++)

https://www.acmicpc.net/problem/1551

1551번: 수열의 변화

---

Level: Bronze I

Solved By: Implementation

 

하나의 배열은 입력받을 배열 li, 다른 하나의 배열은 B[i] = A[i+1] - A[i]를 담을 배열 li_2를 선언하여 과정을 k번 반복하며 li를 li_2로 덮어감으로써 최신화시키는 코드를 작성하였습니다.

 

문제점은 쉽게 파악할 수 있으나, Ternary Operator를 사용함으로써 출력부에서 마지막 인덱싱에서는 ,를 출력하지 않도록 코드를 간단하게 작성한 게 포인트라 생각합니다.

Python

n, k = map(int, input().split())
li = list(map(int, input().split(',')))
li_2 = [0 for _ in range(n)]


for i in range(k): # k번
    for j in range(n-i-1): # B[i] = A[i+1] - A[i]
        li_2[j] = li[j+1] - li[j]
    li = li_2

del li[n-k:len(li)] # 슬라이싱을 통해 원소 삭제

for index, value in enumerate(li):
    print(value, end=',' if index < len(li) - 1 else '')

C++

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
using namespace std;

int n, k;
vector<int> num;

int main(){
    cin >> n >> k;
    string s; cin >> s;
    istringstream iss(s);
    string str_buf;
    while(getline(iss, str_buf, ',')){
        num.push_back(stoi(str_buf));
    }
    
    vector<int> num_;
    for(int i = 0; i < k; i++){
        num_.clear();
        for(int j = 0; j < num.size()-1; j++){
            num_.push_back(num[j+1]-num[j]);
        }
        num = num_;
    }
    
    for(int i = 0; i < num.size(); i++){
        cout << num[i] << (i != num.size()-1 ? ',' : ' ');
    }
    
    return 0;
}