글 읽기 - 아기상어 질문드립니다

superhjj 5년 전 0

테스트 케이스는 모두 맞는데 뭐가 틀린걸까요 고수님들 도와주세용

#include<iostream>
#include<vector>
#include<queue>
using namespace std;

class Feed {
public:
	int x;
	int y;
	int len;
	int d;

	Feed(int x, int y, int len, int d) {
		this->x = x;
		this->y = y;
		this->len = len;
		this->d = d;
	}
};
class cmp {
public:
	bool operator()(const Feed &f1, const Feed &f2) {
		if (f1.d < f2.d) {
			return false;
		}
		else if (f1.d == f2.d) {
			if (f1.x < f2.x) {
				return false;
			}
			else if (f1.x == f2.x) {
				if (f1.y < f2.y) {
					return false;
				}
			}
		}
		return true;
	}
};
int sharkLen = 2;
vector<Feed> feedArr;
int map[20][20];
int N;
int ans = 0;
int dx[] = { -1,0,1,0 };
int dy[] = { 0,1,0,-1 };
int eatCnt = 0;
int visit[20][20];
int dis[20][20];
priority_queue<Feed, vector<Feed>, cmp> que;
queue<pair<int, int>> que1;

void dfs();
bool range(int, int);
void findDis();

int main() {
	cin >> N;

	for (int i = 0; i < N; i++) {
		for (int j = 0; j < N; j++) {
			cin >> map[i][j];
			if (map[i][j] == 9) {
				map[i][j] = 0;
				que1.push(make_pair(i, j));
			}
		}
	}

	dfs();

	cout << ans << endl;
	return 0;
}
void dfs() {
	feedArr.clear();
	if (que.size() != 0) {
		while (!que.empty())
			que.pop();
	}
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < N; j++) {
			visit[i][j] = 0;
			dis[i][j] = 0;
		}
	}
	visit[que1.front().first][que1.front().second] = 1;

	findDis();

	for (int i = 0; i < N; i++) {
		for (int j = 0; j < N; j++) {
			if (map[i][j] >= 1 && map[i][j] <= 6 && sharkLen>map[i][j]) {
				que.push((Feed(i, j, map[i][j], dis[i][j])));
			}
		}
	}

	while (!que.empty()) {
		int x = que.top().x;
		int y = que.top().y;
		int d = que.top().d;
		que.pop();
		//cout << x << " " << y << " " << d << endl;
		map[x][y] = 0;
		eatCnt++;
		if (eatCnt == sharkLen) {
			sharkLen++;
			eatCnt = 0;
		}
		ans += d;
		que1.push(make_pair(x, y));
		dfs();
	}
}
void findDis() {
	while (!que1.empty()) {
		int x = que1.front().first;
		int y = que1.front().second;
		que1.pop();
		for (int i = 0; i < 4; i++) {
			if (visit[x + dx[i]][y + dy[i]] == 0 && range(x + dx[i], y + dy[i])
				&& (sharkLen >= map[x + dx[i]][y + dy[i]])) {
				que1.push(make_pair(x + dx[i], y + dy[i]));
				visit[x + dx[i]][y + dy[i]] = 1;
				dis[x + dx[i]][y + dy[i]] = dis[x][y] + 1;
			}
		}
	}
}
bool range(int x, int y) {
	if (x >= 0 && x < N&&y >= 0 && y < N) {
		return true;
	}
	return false;
}

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

문제

문제

출처

ICPC

아기상어 질문드립니다

superhjj 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말