취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 32980번 문제인 분리배출이다.
문제는 아래 링크를 확인하자.

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

어떤 쓰레기가 한 가지 품목만으로 구성되어 있다면 이 쓰레기는 해당 품목으로 버리거나 일반 쓰레기로 버릴 수 있으며, 그렇지 않은 경우 항상 일반 쓰레기로만 버릴 수 있음을 관찰하자. 특히, 몇몇 일상적인 생각과는 다르게 문제 조건만을 따지면 한 가지 품목만으로 이루어진 쓰레기도 일반 쓰레기로 버려도 된다는 점에 유의하자. 이 경우 두 비용 중 더 저렴한 비용으로 쓰레기를 처리하면 된다. 문제 조건상 일반 쓰레기로 버리는 비용이 재활용 쓰레기로 버리는 비용보다 저렴할 수도 있다는 점에도 유의하자.

어떤 쓰레기가 단일 품목으로 이루어져있는지 여부는 각 인접한 두 문자에 대하여 서로 다른 문자의 쌍이 존재하는지 확인하는 것으로 단순하게 구현할 수 있다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <string>
#include <cstring>
using namespace std;
typedef long long ll;

ll N, ans;
ll cst[128];
ll cnt[128], tmp[128];
string s;
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N;
	while (N--) {
		memset(tmp, 0, sizeof(tmp));
		cin >> s;
		for (auto &l:s) tmp[l]++;
		if (tmp[s.front()] == s.length()) cnt[s.front()] += s.length();
		else cnt['O'] += s.length();
	}
	cin >> cst['P'] >> cst['C'] >> cst['V'] >> cst['S'] >> cst['G'] >> cst['F'] >> cst['O'];
	for (int i = 0; i < 128; i++) ans += min(cst[i], cst['O']) * cnt[i];
	cout << ans;
}