(C++) - LeetCode (easy) 1816. Truncate Sentence

https://leetcode.com/problems/truncate-sentence/

split을 구현하는 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

1. ' '를 구분자로 vector <string>을 반환하는 split을 구현해 vector words에 결과값을 저장합니다.

2. 정답변수 ans를 선언합니다.

📔 풀이과정

k만큼 ans에 words의 원소를 붙여줍니다.

📔 정답 출력 | 반환

 

ans를 반환합니다.

---

📕 Code

📔 C++

class Solution {
public:
  vector <string> split(string input) {
    vector <string> words;
    stringstream ss(input);
    string tmp;
    while(getline(ss, tmp, ' ')) {
      words.push_back(tmp);
    }
    return words;
  }
  string truncateSentence(string s, int k) {
    vector <string> words = split(s);
    string ans = "";
    for(int i = 0; i < k; i++) {
      ans += words[i];
      if(i != k-1) {
        ans += " ";
      }
    }
    return ans;
  }
};

---

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.