글 읽기 - [확인 요청] 16235번 나무 재테크 시간초과 오류 확인 부탁드립니다.

alnimee 4년 전 0

아래 나무재테크 16235번 문제 시간초과로 계속 실패 뜨는데 소스 확인 한 번 부탁드리겠습니다.

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

using namespace std;
const int MAX = 10 + 1;

int N, M, K;
int nutrients[MAX][MAX];
int tmpNutrients[MAX][MAX];
int cntOfTree;

//vector<pair<int, int> > locOfTree;
//vector<vector<int> > Tree;
int locOfTree[MAX][MAX];
vector<int> Tree[MAX][MAX];

int dx[8] = { 0, -1, -1, -1, 0, 1, 1, 1 };
int dy[8] = { 1, 1, 0, -1, -1, -1, 0, 1 };

void Init() {
	cin >> N >> M >> K;
	cntOfTree = M;

	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			cin >> tmpNutrients[i][j];
			nutrients[i][j] = 5;
		}
	}
	for (int i = 0; i < M; i++) {
		int x, y, age;
		cin >> x >> y >> age;
		locOfTree[x][y] = 1;
		Tree[x][y].push_back(age);
	}
}

/*
void Print() {
	for (int i = 0; i < locOfTree.size(); i++) {
		cout << "좌표 x : " << locOfTree[i].first << ", y : " << locOfTree[i].second << " || 해당 좌표의 나무 : ";
		for (int j = 0; j < Tree[i].size(); j++) {
			cout << Tree[i][j] << ' ';
		}
		cout << endl;
	}

	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			cout << nutrients[i][j] << ' ';
		}
		cout << endl;
	}

	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			cout << tmpNutrients[i][j] << ' ';
		}
		cout << endl;
	}
}
*/

void GrowTrees(int x, int y) {
	for (int i = 0; i < 8; i++) {
		int nextX = x + dx[i];
		int nextY = y + dy[i];
		
		if (1 <= nextX && nextX <= N && 1 <= nextY && nextY <= N) {
			locOfTree[nextX][nextY] = 1;
			Tree[nextX][nextY].push_back(1);
			sort(Tree[nextX][nextY].begin(), Tree[nextX][nextY].end());
			cntOfTree++;
			continue;
		}
		else {
			continue;
		}
	}
}

void Winter() {
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			nutrients[i][j] = nutrients[i][j] + tmpNutrients[i][j];
		}
	}
}

void Autumn() {
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			if (locOfTree[i][j] == 0) {
				continue;
			}
			else {
				for (int k = 0; k < Tree[i][j].size(); k++) {
					if (Tree[i][j][k] % 5 == 0) {
						GrowTrees(i, j);
					}
					else {
						continue;
					}
				}
				
			}
		}
	}
}

void Summer(int i, int j, int k) {
	if (k == 0) {
		locOfTree[i][j] = 0;
	}

	for (int l = Tree[i][j].size() - 1; l >= k; l--) {
		nutrients[i][j] = nutrients[i][j] + (Tree[i][j][l] / 2);
		Tree[i][j].pop_back();
		cntOfTree--;
	}
}

void Spring() {
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N; j++) {
			if (locOfTree[i][j] == 0) {
				continue;
			}
			else {
				for (int k = 0; k < Tree[i][j].size(); k++) {
					if (nutrients[i][j] >= Tree[i][j][k]) {
						nutrients[i][j] = nutrients[i][j] - Tree[i][j][k];
						Tree[i][j][k] = Tree[i][j][k] + 1;
					}
					else {
						Summer(i, j, k);
					}
				}
			}
		}
	}
}

void Solve() {
	for (int i = 0; i < K; i++) {
		Spring();
		Autumn();
		Winter();
	}
}

int main(void) {
	Init();
	Solve();
	cout << cntOfTree;
//	Print();
	return 0;
}

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

문제

문제

출처

ICPC

[확인 요청] 16235번 나무 재테크 시간초과 오류 확인 부탁드립니다.

alnimee 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말