검색결과 리스트
마법사 상어와 파이어볼에 해당되는 글 1건
- 2020.11.24 [삼성 SW 역량 테스트] 마법사 상어와 파이어볼 1173
글
[삼성 SW 역량 테스트] 마법사 상어와 파이어볼
삼성 SW 역량 테스트 기출 풀이
2020. 11. 24. 22:58
#include <stdio.h>
#include <vector>
using namespace std;
struct FIREBALL {
int y, x;
int m, s, d;
};
const int dy[8] = { -1, -1, 0, +1, +1, +1, 0, -1 };
const int dx[8] = { 0, +1, +1, +1, 0, -1, -1, -1 };
int N, M, K;
vector<FIREBALL> ball;
vector<int> map[50][50];
void move() {
vector<int> new_map[50][50];
for (int i = 0; i < ball.size(); ++i) {
int dir = ball[i].d;
int speed = (ball[i].s % N);
int ny = (ball[i].y + (dy[dir] * speed) + N) % N;
int nx = (ball[i].x + (dx[dir] * speed) + N) % N;
new_map[ny][nx].push_back(i);
ball[i].y = ny;
ball[i].x = nx;
}
for (int y = 0; y < N; ++y) {
for (int x = 0; x < N; ++x) {
map[y][x] = new_map[y][x];
}
}
}
void sum() {
vector<FIREBALL> new_ball;
for (int y = 0; y < N; ++y) {
for (int x = 0; x < N; ++x) {
if (map[y][x].size() == 0) {
continue;
}
if (map[y][x].size() == 1) {
int index = map[y][x][0];
new_ball.push_back(ball[index]);
continue;
}
int sum_m = 0, sum_s = 0;
bool odd = true, even = true;
for (int i = 0; i < map[y][x].size(); ++i) {
int index = map[y][x][i];
sum_m += ball[index].m;
sum_s += ball[index].s;
if (ball[index].d % 2 == 0) {
odd = false;
}
else {
even = false;
}
}
if (sum_m / 5 == 0) {
continue;
}
int cur_m = sum_m / 5;
int cur_s = sum_s / map[y][x].size();
for (int i = 0; i < 4; ++i) {
if (odd || even) {
new_ball.push_back({ y, x, cur_m, cur_s, i * 2 });
}
else {
new_ball.push_back({ y, x, cur_m, cur_s, i * 2 + 1 });
}
}
}
}
ball = new_ball;
}
int solve() {
while(K--) {
move();
sum();
}
int ret = 0;
for (int i = 0; i < ball.size(); ++i) {
ret += ball[i].m;
}
return ret;
}
int main()
{
scanf("%d %d %d", &N, &M, &K);
for (int i = 0; i < M; ++i) {
int y, x, m, s, d;
scanf("%d %d %d %d %d", &y, &x, &m, &s, &d);
--y, --x;
ball.push_back({ y, x, m, s, d });
map[y][x].push_back(i);
}
int ret = solve();
printf("%d\n", ret);
return 0;
}