백준 단계 5 : 문자열 (C++)

27866. 단어 S와 정수 i가 주어졌을 때, S의 i번째 글자를 출력하는 프로그램을 작성하시오.

#include<iostream>
#include <string>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    int i;
    cin >> s >> i;
    cout << s[i-1];
    return 0;
}

string을 사용하여 해당 인덱스를 추출하면 된다. 

몰랐던 사실엔데 #include <iostream>에는 #include <string>이 포함되어있어서 굳이 string 을 include 해주지 않아도 정상적으로 적당한다

 

 

2743. 알파벳으로만 이루어진 단어를 입력받아, 그 길이를 출력하는 프로그램을 작성하시오.

#include<iostream>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    cin >> s;
    cout << s.length();
    return 0;
}

문자열 길이는 length 함수로 구할 수 있다. size() 함수로도 구할 수 있다. 값은 동일하다.

 

 

9086. 문자열을 입력으로 주면 문자열의 첫 글자와 마지막 글자를 출력하는 프로그램을 작성하시오.

#include<iostream>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;
    string s;
    cin >> t;
    for (int i = 0; i < t; i++)
    {
        cin >> s;
        cout << s[0] << s[s.length()-1] << "\n";
    }
    return 0;
}

for문 대신 while(n--)와 같은 방식도 많이 사용하는 것으로 보인다.

 

 

11654. 알파벳 소문자, 대문자, 숫자 0-9중 하나가 주어졌을 때, 주어진 글자의 아스키 코드값을 출력하는 프로그램을 작성하시오.

#include<iostream>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    char c;
    cin >> c;
    cout << int(c);
    return 0;
}

char 타입을 int로 바꾸어주면 된다.

 

 

11720. N개의 숫자가 공백 없이 쓰여있다. 이 숫자를 모두 합해서 출력하는 프로그램을 작성하시오.

#include<iostream>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin >> n;
    char* c = new char[n+1];
    cin >> c;
    int sum = 0;
    for (int i = 0; i < n; i++)
    {
        sum += int(c[i]) - '0';
    }
    cout << sum << '\n';
}

배열을 사용하지 않고 for 문에 cin을 넣는 방법도 있었다. 혹은 string으로 받고 추출하는 방법도 있다.

 

 

10809.  알파벳 소문자로만 이루어진 단어 S가 주어진다. 각각의 알파벳에 대해서, 단어에 포함되어 있는 경우에는 처음 등장하는 위치를, 포함되어 있지 않은 경우에는 -1을 출력하는 프로그램을 작성하시오.

#include<iostream>
#include<string.h>
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    cin >> s;
    int n = s.length();
    int location[26];
    for (int i = 0; i < 26; i++)
    {
        location[i] = -1;
    }
    for (int i = 0; i < s.length(); i++) {
        if (location[(s[i] - 'a')] == -1)
            location[(s[i] - 'a')] = i;
    }
    for (int i = 0; i < 26; i++)
    {
        cout << location[i] << " ";
    }
}

하나하나 찾고 새는 방식으로 작성하였다. 하지만 find 함수를 사용하면 아주 간단하게 풀 수 있다.

string str = "abcd";
cout << str.find("d");

a부터 0이 되므로 출력값은 3이 된다. str.find("d", 찾기시작할 index);로 변형하여 사용할 수도 있다. 찾는 문자가 없을 경우 -1을 return 하게 된다.

#include <iostream>
using namespace std;
int main() {
	string str;
	string alphabet = "abcdefghijklmnopqrstuvwxyz";
	cin >> str;
	for (int i = 0; i < alphabet.length(); i++)
		cout << (int)str.find(alphabet[i]) << " ";
	return 0;
}

find 앞에 (int)를 붙여 변환하는 이유는 해당 값이 없을때 npos를 반환하게 되는데, npos는 unsigned int이다. int로 변환했을 때 대응되는 값이 -1이 된다. find 함수에 커서를 대보면 size_t가 뜨는데, size_t는 일반적으로 unsigned long long이다. string의 find 함수를 사용하여 구현하는 것이 훨씬 간단하다.

 

 

2675. 문자열 S를 입력받은 후에, 각 문자를 R번 반복해 새 문자열 P를 만든 후 출력하는 프로그램을 작성하시오. 즉, 첫 번째 문자를 R번 반복하고, 두 번째 문자를 R번 반복하는 식으로 P를 만들면 된다. S에는 QR Code "alphanumeric" 문자만 들어있다.

#include <iostream>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	//cin.tie(0);
	int t, r;
	string s;
	cin >> t;
	while (t--) {
		cin >> r >> s;
		for (int i = 0; i < s.length(); i++)
		{
			for (int j = 0; j < r; j++)
			{
				cout << s[i];
			}
		}
		cout << "\n";
	}
	return 0;
}

이중 for문을 사용하여 구현하였다. 앞서 말했듯이 문자열 길이는 length 함수로 구할 수 있다. size() 함수로도 구할 수 있다. 

 

 

1152. 영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.

string sen;
cin >> sen;
cout << sen.length() << endl;

The Curious Case of Benjamin Button 를 입력해본 결과 'The'만 sen에 저장되어 3이 출력되었다. 공백을 포함한 문자열을 받는 방법을 몰라서 함수를 찾아보았다. 바로 getline을 사용해주면 된다.

string str;
getline(cin, str);
cout << str;
cout << str[0];

getline을 사용하기 위해서는 #include<string>을 꼭! 해주어야 한다. str을 출력하면 The Curious Case of Benjamin Button가 출력된다.

#include <iostream>
#include <string>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string sen;
	getline(cin, sen);
	int count = 1;
	if ((sen.length() == 1) && (sen[0] == ' ')) {
		count = 0;
	}
	else {
		for (int i = 1; i < sen.length() - 1; i++)
		{
			if (sen[i] == ' ')
				count++;
		}
	}
	cout << count;
	return 0;
}

처음 문자는 공백이든, 실제 문제이든 신경쓸 필요가 없다. 만약 두번째 글자가 abc와 같은 값이라면 앞에 뭐가 오든 첫 문단의 일부 또는 첫 문자가 될 것이며 두번째 글자가 공백이라면 첫번째 글자가 abc와 같은 값이므로 첫번째 문단을 count하게 된다.

다른 코드를 봤는데 sen.empty() 를 사용하여 코드를 구현하는 경우도 있었다. 하지만 영어 대소문자와 공백으로 이루어진 문자라고 주어줬기 때문에 아무런 문자 없는 경우는 고려하지 않아도 된다. empty() 함수는 정말 아무것도 입력하지 않았을 경우만 해당된다. 그냥 공백으로만 이루어진 경우만 고려해주면 된다.

 

 

2908. 상근이의 동생 상수는 수학을 정말 못한다. 상수는 숫자를 읽는데 문제가 있다. 이렇게 수학을 못하는 상수를 위해서 상근이는 수의 크기를 비교하는 문제를 내주었다. 상근이는 세 자리 수 두 개를 칠판에 써주었다. 그 다음에 크기가 큰 수를 말해보라고 했다. 상수는 수를 다른 사람과 다르게 거꾸로 읽는다. 예를 들어, 734와 893을 칠판에 적었다면, 상수는 이 수를 437과 398로 읽는다. 따라서, 상수는 두 수중 큰 수인 437을 큰 수라고 말할 것이다. 두 수가 주어졌을 때, 상수의 대답을 출력하는 프로그램을 작성하시오.

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string num1, num2;
	cin >> num1 >> num2;
	swap(num1[0], num1[2]);
	swap(num2[0], num2[2]);
	if (num1 > num2) {
		cout << num1;
	}
	else {
		cout << num2;
	}
	return 0;
}

algorithm 헤더파일의 swap() 함수를 사용하였다. swap 함수 없이 뒤에서부터 비교하는 방식으로 구현해도 된다. 두 문자열 모두 숫자로만 되어있기 때문에 크기를 비교할 때 그냥 비교해도 된다. 

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string num1, num2;
	cin >> num1 >> num2;
	swap(num1[0], num1[2]);
	swap(num2[0], num2[2]);
	cout << max(num1, num2);
	return 0;
}

이렇게 직접 max 함수를 사용해도 괜찮다. num1[0]과 num1[2]가 아니라 num1.begin()와 num1.end() 같이 시작과 끝을 추출해서 할려면 reverse 함수를 사용하면 된다. reverse(begin(a), end(a)) 와 같이 적어도 된다.

 

 

 

5622. 상근이의 할머니는 아래 그림과 같이 오래된 다이얼 전화기를 사용한다. 전화를 걸고 싶은 번호가 있다면, 숫자를 하나를 누른 다음에 금속 핀이 있는 곳 까지 시계방향으로 돌려야 한다. 숫자를 하나 누르면 다이얼이 처음 위치로 돌아가고, 다음 숫자를 누르려면 다이얼을 처음 위치에서 다시 돌려야 한다. 숫자 1을 걸려면 총 2초가 필요하다. 1보다 큰 수를 거는데 걸리는 시간은 이보다 더 걸리며, 한 칸 옆에 있는 숫자를 걸기 위해선 1초씩 더 걸린다.

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string num;
	int time = 0;
	cin >> num;
	for (int i = 0; i < num.length(); i++)
	{
		if (num[i] <= 'C')
			time += 3;
		else if (num[i] <= 'F')
			time += 4;
		else if (num[i] <= 'I')
			time += 5;
		else if (num[i] <= 'L')
			time += 6;
		else if (num[i] <= 'O')
			time += 7;
		else if (num[i] <= 'S')
			time += 8;
		else if (num[i] <= 'V')
			time += 9;
		else
			time += 10;
	}
	
	cout << time;

	return 0;
}

내가 처음으로 짠 코드는 if 문으로 비교하는 코드였다.

 

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string num;
	cin >> num;
	int time[] = { 3,3,3,4,4,4,5,5,5,6,6,6,7,7,7,8,8,8,8,9,9,9,10,10,10,10 };
	int result = 0;

	for (int i = 0; i < num.length(); i++)
	{
		result += time[(num[i] - 'A')];
	}
	cout << result;
	return 0;
}

이와 같이 배열을 사용하면 코드 길이를 줄일 수 있다. 아스키코드를 사용하여 index로 변환하고 이를 이용하는 코드이다.

 

 

11718. 입력 받은 대로 출력하는 프로그램을 작성하시오.

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	string sen;
	while (1) {
		getline(cin, sen);
		if (sen == "\0")
			break;
		cout << sen << '\n';
	}
	return 0;
}

앞에서 사용한 getline 함수를 사용하여 공백 문자도 받아오도록 한다.

문장의 끝에는 자동으로 NULL 문자가 저장이 된다. NULL은 cin.tie(0)에서 한 것처럼 0으로도 표현할 수 있는데, 숫자 0이랑 구분하기 위해 \0으로 표현한다. 혹은 sen.length() == 0 으로 해도 된다. 또한 sen == ""와 같이 표현할 수도 있다.

 

while (getline(cin, sen)) {
	cout << sen << endl;
}

앞에서 배운 EOF 를 사용하여 하는 방법도 있다. 

 

https://shine-slowly.tistory.com/25

백준 단계 3 : 반복문 (C++)

eof는 위 링크의 10952번을 확인하면 된다.