백준 알고리즘 [1152번] (C++) 단어의 개수

---

Code

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

int Word_Num(vector<string> &a, string &b);

int main()
{
    ios::sync_with_stdio(NULL);
    cin.tellg();

    vector<string> str;
    string s;

    getline(cin, s);

    str.push_back(s);
    cout << Word_Num(str, s)+1 << "\n";
}
int Word_Num(vector<string> &a, string &b)
{
    int count = 0;
    for (int i = 0; i < a[0].length(); i++)
        if (a[0][i] == ' ')
            count++;
    if (a[0][0] == ' ')
        count--;
    if (a[0][b.length() - 1] == ' ')
        count--;
    return count;
}