나무 재테크

https://www.acmicpc.net/problem/16235

16235번: 나무 재테크

 

실패 한 이유

처음에는 info 라는 구조체를 만들어서 했다.

비효율적이다.

vector<int> map[15][15];

로 해서

 

vector<int> tmp;

sort(map[i][j].begin(), map[i][j].end());

map[i][j].clear();

for (int k = 0; k < tmp.size(); k++)

{

    map[i][j].push_back(tmp[k]);

}

 

1. 사계를 fourseason으로 묶어서 했었다.

2. memset(nut, 5, sizeof(nut)) 얘가 문제였다.

memset에는 0(false), 1(true), 0x3f 만 들어갈 수 있다.

 

<소스 코드>

#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
#include <algorithm>
 
using namespace std;
 
int dx[] = { -1, -1, -1, 0, 0, 1, 1, 1 };
int dy[] = { -1, 0, 1, -1, 1, -1, 0, 1 };
 
// 하나의 칸에 여러개의 나무
vector<int> map[15][15];
 
int nut[15][15];
int input[15][15];
 
int n, m, k;
 
void breed(int row, int col)
{
    for (int k = 0; k < 8; k++)
    {
        int nx = row + dx[k];
        int ny = col + dy[k];
        if (nx >= 1 && nx <= n && ny >= 1 && ny <= n)
        {
            map[nx][ny].push_back(1);
        }
    }
}
 
void harunatsu()
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            if (map[i][j].size() == 0)
                continue;
            else
            {
                int die_yang = 0;
                // 산 나무들을 tmp에 저장
                vector<int> tmp;
                // 나이 오름차순 정렬
                sort(map[i][j].begin(), map[i][j].end());
                for (int a = 0; a < map[i][j].size(); a++)
                {
                    int age = map[i][j][a];
                    if (age <= nut[i][j])
                    {
                        nut[i][j] -= map[i][j][a];
                        tmp.push_back(age + 1);
                    }
                    // 양분이 된다
                    else
                        die_yang += (map[i][j][a] / 2);
                }
 
                map[i][j].clear();
                for (int k = 0; k < tmp.size(); k++)
                {
                    map[i][j].push_back(tmp[k]);
                }
 
                nut[i][j] += die_yang;
            }
        }
    }
}
 
void aki()
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            if (map[i][j].size() == 0)
                continue;
            else
            {
                for (int a = 0; a < map[i][j].size(); a++)
                {
                    // 나이가 5의 배수라면
                    if (map[i][j][a] % 5 == 0)
                        breed(i, j);
                }
            }
        }
    }
}
 
void fuyu()
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            nut[i][j] += input[i][j];
        }
    }
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    cin >> n >> m >> k;
 
    // 얘가 문제였음
    // memset에는 5를 넣을수 없거든!
    // memset(nut, 5, sizeof(nut));
 
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            cin >> input[i][j];
            nut[i][j] = 5;
        }
    }
 
    for (int i = 0; i < m; i++)
    {
        int x, y, toshi;
        cin >> x >> y >> toshi;
        map[x][y].push_back(toshi);
    }
 
    for (int i = 0; i < k; i++)
    {
        harunatsu();
        aki();
        fuyu();
    }
 
    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            ans += map[i][j].size();
        }
    }
 
    cout << ans << '\n';
    return 0;
}