본문 바로가기
알고리즘/백준

[백준 2257] 화학식량 (C++)

by fortissimo 2024. 11. 13.

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

문제


우리가 널리 사용하는 H2O(물), CH3COOH(아세트산)과 같은 화학식은 알파벳과 숫자, 그리고 괄호로 구성된다. 먼저 알파벳은 원자를 나타내는 것으로 H는 수소(Hydrogen), C는 탄소(Carbon), O는 산소(Oxygen) 원자를 뜻한다. 또한 원자를 나타내는 알파벳 뒤에 따르는 숫자는 그 원자가 몇 개 포함되어 있는지를 뜻한다. 따라서 COOHHH 분자는 CO2H3로 나타낼 수 있다. 이 문제에서, 숫자는 항상 2 이상 9 이하로만 입력으로 주어진다. 따라서 CO23과 같이 숫자가 두자리인 경우는 없다.

물의 화학식을 보고 물은 두 개의 수소 원자와 한 개의 산소 원자로 이루어졌음을 알 수 있다. 또한 아세트산의 화학식처럼 한 종류의 알파벳이 화학식에 여러 번 나타날 수도 있다. 실제 화학식 또한 이렇게 사용되는데, 이는 분자의 결합 구조를 나타내기 위함이다.

종종 화학식에는 괄호가 사용되기도 하는데 괄호로 묶인 원자들은 하나의 새로운 원자와 같은 작용을 한다. 따라서 CH(CO2H)(CO2H)(CO2H) 분자는 CH(CO2H)3와 같이 나타낼 수 있다. 괄호 안에 아무런 알파벳도 없는 경우도 있을 수 있는데, 이런 경우는 괄호가 없는 경우와 마찬가지라고 생각하면 된다.

이러한 화학식을 보고 우리는 화학식량을 계산할 수 있는데, 화학식량이란 그 화학식에 포함되어 있는 모든 원자들의 질량의 합을 말한다. 수소 원자 하나의 질량은 1, 탄소 원자 하나의 질량은 12, 산소 원자 하나의 질량은 16이다. 물은 두 개의 수소 원자와 한 개의 산소 원자로 이루어져 있으므로 물의 화학식량은 18이다.

화학식이 주어졌을 때, 이 화학식의 화학식량을 계산하는 프로그램을 작성하시오. 화학식은 수소, 탄소, 산소만을 포함하고 있는 것만이 입력으로 주어진다.

 

입력


첫째 줄에 화학식이 주어진다. 화학식은 H, C, O, (, ), 2, 3, 4, 5, 6, 7, 8, 9만으로 이루어진 문자열이며, 그 길이는 100을 넘지 않는다.

 

출력


첫째 줄에 화학식량을 출력한다. 분자량이 10,000이 넘는 고분자는 입력으로 주어지지 않는다.

 

문제 풀이


스택 문제.

 

)를 만나면 (전에 등장한 문자열에 대해 계산해주면 된다.

계산을 위해 문자를 저장하는 alphabets 스택과 계산 중인 결과를 저장하는 numbers 스택을 만들었다. )를 제외한 문자가 나오면 alphabets에 저장하고, )가 나오면 (가 나오기 전까지 numbers에 있는 숫자들을 순차적으로 더해주면 된다. 아래 코드의 계산 방식은 괄호 안에 있는 C, H, O의 개수만큼 numbers의 수를 더하는 식으로 구현되었다. 따라서 괄호 안에 존재하는 알파벳의 수와 계산 대기 중인 numbers 안의 개수가 같아야 한다. 한 번 괄호 안에 있는 수들을 모두 계산하면 numbers 안의 수와 달라지기 때문에, 아무 문자나 alphabets에 넣어 계산 결과가 끝났음을 알린다.

 

아래는 코드.

#include <iostream>
#include <stack>
using namespace std;

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

	string str;
	cin >> str;
	stack<char> alphabets;
	stack<int> numbers;
	for (int i = 0; i < str.length(); i++)
	{
		if (str[i] == ')')
		{
			int sum = 0;
			while (alphabets.top() != '(')
			{
				sum += numbers.top();
				numbers.pop();
				alphabets.pop();
			}
			alphabets.pop();
			alphabets.push('X');
			numbers.push(sum);
		}
		else if (str[i] >= 48 && str[i] <= 57)
		{
			int num = numbers.top();
			numbers.pop();
			num *= str[i] - 48;
			numbers.push(num);
		}
		else
		{
			alphabets.push(str[i]);
			if (alphabets.top() == 'C')
			{
				numbers.push(12);
			}
			else if (alphabets.top() == 'O')
			{
				numbers.push(16);
			}
			else if (alphabets.top() == 'H')
			{
				numbers.push(1);
			}
		}
	}
	int sum = numbers.top();
	numbers.pop();
	while (!numbers.empty())
	{
		sum += numbers.top();
		numbers.pop();
	}
	cout <<sum<< "\n";
	return 0;
}

 

'알고리즘 > 백준' 카테고리의 다른 글

[백준 11505] 구간 곱 구하기 (C++)  (0) 2024.11.16
[백준 13334] 철로 (C++)  (2) 2024.11.15
[백준 12873] 기념품 (C++)  (0) 2024.11.12
[백준 1110] 더하기 사이클 (C++)  (0) 2024.11.10
[백준 23559] 밥 (C++)  (0) 2024.11.09