백준 1244번 스위치 켜고 끄기

문제 링크입니다: https://www.acmicpc.net/problem/1244

1244번: 스위치 켜고 끄기

문제에서 주어진 조건대로 코드를 작성하면 되는 문제였습니다.

보통 여학생이 주어졌을 때 헤매셨을 것이라고 생각하는데 for문을 적절히 사용하시면 간단하게 문제를 해결할 수 있습니다.

 

	#include <iostream>
	#include <vector>
	#include <algorithm>
	using namespace std;
	const int MAX_INDEX = 20;
	const int MAX = 100 + 1;
	int N;
	bool switches[MAX];
	void pressSwitch(int idx)
	{
	switches[idx] = 1 - switches[idx];
	}
	bool isEndOfLine(int idx)
	{
	return (idx % MAX_INDEX == 0 || idx == N);
	}
	int main(void)
	{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cin >> N;
	for (int i = 1; i <= N; i++)
	{
	cin >> switches[i];
	}
	int inputs;
	cin >> inputs;
	for (int i = 0; i < inputs; i++)
	{
	int gender, num;
	cin >> gender >> num;
	if (gender == 1)
	{
	for (int j = num; j <= N; j += num)
	{
	pressSwitch(j);
	}
	continue;
	}
	pressSwitch(num);
	for (int j = num - 1, k = num + 1; (j >= 1 && k <= N && switches[j] == switches[k]); j--, k++)
	{
	pressSwitch(j);
	pressSwitch(k);
	}
	}
	for (int i = 1; i <= N; i++)
	{
	cout << switches[i];
	if (isEndOfLine(i))
	{
	cout << "\n";
	continue;
	}
	cout << " ";
	}
	return 0;
	}

view raw .cpp hosted with ❤ by GitHub

개발환경:Visual Studio 2017

 

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~