취미로 코딩하기

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

이번에 볼 문제는 백준 1034번 문제인 램프이다.
문제는 아래 링크를 확인하자.

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

이 문제를 해결하는 데에 가장 중요한 관찰은 다음과 같다:

어떤 두 행의 램프가 동시에 불이 모두 켜진다면, 처음 주어진 두 행의 상태도 똑같아야 한다.

위의 관찰을 기반으로, multiset 자료구조를 이용하여 같은 행을 먼저 모으고 각 행별로 주어진 횟수 K로 불을 전부 킬 수 있는지 확인하는 것으로 문제를 해결할 수 있다.

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

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

multiset<string> mltset;

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

	int N, M; cin >> N >> M;
	while (N--) {
		string s; cin >> s;
		mltset.insert(s);
	}

	int K; cin >> K;
	int ans = 0;
	for (auto row : mltset) {
		int cnt = 0;
		for (auto lamp : row) {
			if (lamp == '0') cnt++;
		}
		if (cnt <= K && !((K - cnt) & 1)) ans = max(ans, (int)mltset.count(row));
	}

	cout << ans;
}