• 카테고리

    질문 & 답변
  • 세부 분야

    알고리즘 · 자료구조

  • 해결 여부

    해결됨

46번 문제 질문 있습니다!

20.04.04 03:00 작성 조회수 91

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;
}