인프런 커뮤니티 질문&답변

공부왕님의 프로필 이미지
공부왕

작성한 질문수

it 취업을 위한 알고리즘 문제풀이 입문 (with C/C++) : 코딩테스트 대비

46. 멀티 태스킹

46번 문제 질문 있습니다!

해결된 질문

작성

·

111

0

안녕하세요 46번 다시 풀다가 5번째 2000개 데이터 input에서 틀려서 궁금해서 질문합니다.

K초 작업이 끝나고 -> 다음 index인 값이 0이면 0이 아닌 값을 만날때까지 ++한다음에 답을 출력

하게 짰는데요!(영상에서 설명하신 거처럼)
근데 왜 계속 마지막에서 598이라는 답이 나오는 지 궁금합니다. (559가 정답입니다!)
제가 놓치고 있는 부분이 있을 까요?

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

int a[2001];

int main() {
	int N, K, total = 0;
	scanf("%d", &N);
	for (int i = 0; i < N; i++) {
		scanf("%d", &a[i]);
		total += a[i];
	}
	scanf("%d", &K);
	if (total <= K) {
		printf("-1");
		exit(0);
	}
	int index = 0;
	while (K--) {
		if (index == N) index = 0;
		if(a[index] !=0 )a[index]--;
		else {
			while (a[index] == 0) {
				if (index == N) index = 0;
				index++;
			}
			a[index]--;
		}
		index++;
	}
	if (a[index] == 0) {
		while (a[index] == 0) {
			if (index == N) index = 0;
			index++;
		}
	}
	printf("%d\n", index+1);

	return 0;
}

답변 1

1

김태원님의 프로필 이미지
김태원
지식공유자

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

int a[2000];

int main() {
	freopen("input.txt", "rt", stdin);
	int N, K, total = 0;
	scanf("%d", &N);
	for (int i = 0; i < N; i++) {
		scanf("%d", &a[i]);
		total += a[i];
	}
	scanf("%d", &K);
	if (total <= K) {
		printf("-1");
		exit(0);
	}
	int index = 0;
	while (K--) {
		if (index == N) index = 0;
		if(a[index] !=0 )a[index]--;
		else {
			while (a[index] == 0) {
				index++;
				if (index == N) index = 0;
				//index++;      여기에 있으면 안되는 이유 : while 문이 시작할 때 index가 N-1값이라면 a[index]값은 당연히 0이겠죠. 그럼 a[N]의 값을 탐색하게 됩니다. 
			}
			a[index]--;
		}
		index++;
	}
	if (a[index] == 0) {
		while (a[index] == 0) {
			if (index == N) index = 0;
			index++;
		}
	}
	printf("%d\n", index+1);

	return 0;
}
공부왕님의 프로필 이미지
공부왕

작성한 질문수

질문하기