글 읽기 - 반례부탁드립니다 ㅜㅜ

kimtg3 4년 전 0

반례좀 부탁드리겠습니다 ㅜㅜ
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <queue>
#include <cstdio>
using namespace std;

int n, m, t; // n개의원판, m:숫자갯수, t번회전
int x, d, k; // 번호가 x배수인 원판을 d방향으로 k칸회전
int dy[4] = { 1,-1,0,0 };
int dx[4] = { 0,0,-1,1 };
int arr[51][51]; // 원판
int c_arr[51][51]; // 복사본
int visited[51][51];
int answer, n_cnt;
int cnt = 1;
bool flag;
vector <pair<int, int> > v;

void copy_map(int A[51][51], int B[51][51]) { // 배열복사
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j < m; j++) {
            A[i][j] = B[i][j];
        }
    }
}

void rotate(int idx, int num) { // idx원판을 시계방향으로 num칸 회전
    for (int i = 0; i < m; i++) { // num칸 이동
        arr[idx][(i + num) % m] = c_arr[idx][i];
    }
    for (int i = 0; i < m; i++) { // 복사
        c_arr[idx][i] = arr[idx][i];
    }    
}

void dfs(int y, int x, int target) {
    visited[y][x] = 1;
    for (int i = 0; i < 4; i++) {
        int ny = y + dy[i];
        int nx = x + dx[i];
        if (nx < 0) nx = m - 1;
        if (nx >= m) nx = 0;

        if (visited[ny][nx] == 1 || c_arr[ny][nx] != target || c_arr[ny][nx] == 0) continue; // 방문했었고 인접해있지 않으면 continue
        if (ny < 1 || ny > n) continue;

        v.push_back({ ny,nx }); // 인접한수들
        flag = true; // 인접한수들 있으면 true로 표시
        dfs(ny, nx, target);
    }
}


int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    cin >> n >> m >> t;
    n_cnt = n * m;

    for (int i = 1; i <= n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> arr[i][j];
            c_arr[i][j] = arr[i][j];
            answer += arr[i][j];
        }
    }

    while (t--) {
        memset(visited, 0, sizeof(visited));
        flag = false;
        cin >> x >> d >> k;

        int cnt = 1;
        while (x * cnt <= n) {
            if (d == 0) { // 시계방향
                rotate(x * cnt, k);
            }
            else { // 반시계 k방향 = 시계 m-k방향
                rotate(x * cnt, m - k);
            }
            cnt++;
        }

        for (int i = 1; i <= n; i++) {
            for (int j = 0; j < m; j++) {
                if (!visited[i][j] && arr[i][j] != 0) {
                    v.clear();
                    v.push_back({ i,j });
                    dfs(i, j, arr[i][j]); // 인접하면서 같은수 탐색
                    if (v.size() > 1) { // 인접한수 있을경우 숫자 삭제
                        for (int a = 0; a < v.size(); a++) {
                            answer -= arr[v[a].first][v[a].second];
                            arr[v[a].first][v[a].second] = 0;
                            copy_map(c_arr, arr);
                            n_cnt--;
                        }
                    }
                }
            }
        }

        double aver;
        if (!flag && n_cnt != 0) { // 인접한 수 없을경우 평균구하고 +-
            aver = (double)answer / (double) n_cnt;
            for (int i = 1; i <= n; i++) {
                for (int j = 0; j < m; j++) {
                    if (arr[i][j] == 0) continue;
                    if (aver > (double)arr[i][j]) {
                        arr[i][j]++;
                    }
                    else if (aver < (double)arr[i][j]){
                        arr[i][j]--;
                    }
                }
            }
            copy_map(c_arr, arr);
        }
        
    }

    cout << answer << '\n';

    return 0;
}
댓글을 작성하려면 로그인해야 합니다.
문제

문제

출처

ICPC

반례부탁드립니다 ㅜㅜ

kimtg3 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말
