글 읽기 - c++(주석) 틀린 부분을 모르겠습니다 ㅠㅠ

kjw2448 3년 전 0

예제랑 질문 반례는 다 맞는거 같은데 틀렸다고 나오네요 ㅠㅠ

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include <cstring>

using namespace std;

// 상어 정보
struct shark {
	int size;
	int speed;
	int dir;
};

bool compare(shark a, shark b) {
	return a.size < b.size;
}

int n, m, shark_num;
int person = -1;
int dx[4] = { -1,1,0,0 };
int dy[4] = { 0,0,1,-1 };
int ans;
bool vis[101][101];


void func(vector<vector<vector<shark>>> &map) {
	while (true) {
		person++;
		// person 열이 벗어나면 return;
		if (person >= m) return;

		// 땅과 가장 가까운 상어가 있으면 ans +=;
		for (int i = 0; i < n; i++) {
			if (map[i][person].size() != 0) {
				ans += map[i][person][0].size;
				map[i][person].erase(map[i][person].begin());

				break;
			}
		}

		// 상어 이동 정보를 tmp에 저장
		vector<vector<vector<shark>>> tmp(n, vector<vector<shark>>(m, vector<shark>()));
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (map[i][j].size() == 0) continue;
				if (map[i][j].size() == 1 && vis[i][j]) continue;
				
				int x = i;
				int y = j;
				int d = map[i][j][0].dir;
				int s = map[i][j][0].size;
				int sp = map[i][j][0].speed;

				
				for (int k = 0; k < sp; k++) {
					x = x + dx[d];
					y = y + dy[d];
					if (x < 0 || x >= n || y < 0 || y >= m) {
						if (d == 0) {
							x = x + 2;
							d = 1;
						}
						else if (d == 1) {
							x = x - 2;
							d = 0;
						}
						else if (d == 2) {
							y = y - 2;
							d = 3;
						}
						else {
							y = y + 2;
							d = 2;
						}
					}
				}
				vis[x][y] = 1;
				tmp[x][y].push_back({ s,sp,d });
			}
		}
		memset(vis, false, sizeof(vis));

		//같은 곳에 있는 상어가 있으면 size 오름차순으로 정리 후 마지막만 남기고 삭제
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (tmp[i][j].size() > 1) {
					sort(tmp[i][j].begin(), tmp[i][j].end(), compare);
					while (tmp[i][j].size() != 1) tmp[i][j].erase(tmp[i][j].begin());
				}
			}
		}
		map = tmp;
	}
}

int main() {
	cin >> n >> m >> shark_num;
	//
	vector<vector<vector<shark>>> map(n, vector<vector<shark>>(m,vector<shark>()));
	for (int i = 0; i < shark_num; i++) {
		int r, c, s, d, z;
		cin >> r >> c >> s >> d >> z;
		map[r - 1][c - 1].push_back({ z,s,d-1 });
	}

	func(map);

	cout << ans;
}

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

문제

문제

출처

ICPC

c++(주석) 틀린 부분을 모르겠습니다 ㅠㅠ

kjw2448 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말