글 읽기 - 이게 왜 틀렸는지 도움 부탁드리겠습니다!

tprjs456 2년 전 0

제가 찾았던 반례들을 검색해보면 다 옳은 답이 나와서 제출하면 틀렸다고 나왔습니다.

아무리 생각해도 어떤 부분이 오류가 나는건지 찾지 못했습니다.

제 코드에서 고쳐야될점이나 틀린점 도움 부탁드립니다.

#include<iostream>
#include<vector>
#include<algorithm>
#include<climits>
#include<cmath>
#define MIN -11111
using namespace std;
int N, M, K;
vector<vector<int>> vec;
int result = 0;
int main(void) {
	cin >> N >> M >> K;
	for (int i = 0; i < N; i++) {
		vector<int> _vec;
		for (int j = 0; j < M; j++) {
			int a;
			cin >> a;
			_vec.push_back(a);
		}
		vec.push_back(_vec);
	}
	

		for (int k = 0; k < K; k++) {
			int arr[4] = { MIN,MIN,MIN,MIN };
			int x, y;
			int max = MIN;
			int max_a = MIN;

			for (int i = 0; i < N; i++) {		//최댓값
				for (int j = 0; j < M; j++) {
					if (vec[i][j] > max) { max = vec[i][j]; x = i; y = j; }
				}
			}

			vec[x][y] = MIN;
			if (x - 1 >= 0) { arr[0] = vec[x - 1][y]; vec[x - 1][y] = MIN; }//첫번째로 고른 최대값의 주변합의 최대
			if (x + 1 < N) { arr[1] = vec[x + 1][y]; vec[x + 1][y] = MIN; }
			if (y - 1 >= 0) { arr[2] = vec[x][y - 1]; vec[x][y - 1] = MIN; }
			if (y + 1 < M) { arr[3] = vec[x][y + 1]; vec[x][y + 1] = MIN; }
			int sum = MIN * 2;
			for (int i = 0; i < 3; i++) {
				for (int j = i + 1; j < 4; j++) {
					if (arr[i] != MIN && arr[j] != MIN) {
						if (arr[i] + arr[j] > sum) { sum = arr[i] + arr[j]; }
					}
				}
			}

			for (int i = 0; i < N; i++) {		//다음번의 예상 최댓값을 구함
				for (int j = 0; j < M; j++) {
					if (max_a < vec[i][j]) { max_a = vec[i][j]; }
				}
			}
			if (max + max_a < sum) {		//주변합들의 최대와 다음에 나올 최댓값을 비교
				if (arr[0] != MIN) { vec[x - 1][y] = arr[0]; }
				if (arr[1] != MIN) { vec[x + 1][y] = arr[1]; }
				if (arr[2] != MIN) { vec[x][y - 1] = arr[2]; }
				if (arr[3] != MIN) { vec[x][y + 1] = arr[3]; }
				k--;
			}
			else {
				result += max;
			}
			max = MIN;
			max_a = MIN;
		}
		cout << result;
	
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

이게 왜 틀렸는지 도움 부탁드리겠습니다!

tprjs456 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말