Doby's Lab

[알고리즘] 백준 17829번: 222-풀링 (C++) 본문

PS/BOJ

[알고리즘] 백준 17829번: 222-풀링 (C++)

도비(Doby) 2022. 2. 26. 15:50

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

 

17829번: 222-풀링

조기 졸업을 꿈꾸는 종욱이는 요즘 핫한 딥러닝을 공부하던 중, 이미지 처리에 흔히 쓰이는 합성곱 신경망(Convolutional Neural Network, CNN)의 풀링 연산에 영감을 받아 자신만의 풀링을 만들고 이를 22

www.acmicpc.net

오랜만에 푼 분할 정복 문제였다.

헷갈렸던 부분은 함수를 구현할 때였는데

int dq(int num, int i, int j){
    if(num == 1){
        return arr[i][j];
    }
    
    int temp[5];
    temp[1] = dq(num / 2, i, j);
    temp[2] = dq(num / 2, i + num / 2, j);
    temp[3] = dq(num / 2, i, j + num / 2);
    temp[4] = dq(num / 2, i + num / 2, j + num / 2);
    
    sort(temp + 1, temp + 5);
    return temp[3];
}

이렇게 구현했어야 하는 걸

int dq(int num, int i, int j){
    if(num == 1){
        return arr[i][j];
    }
    
    int temp[5];
    temp[1] = dq(num / 2, 1, 1);
    temp[2] = dq(num / 2, 1 + num / 2, 1);
    temp[3] = dq(num / 2, 1, 1 + num / 2);
    temp[4] = dq(num / 2, 1 + num / 2, 1 + num / 2);
    
    sort(temp + 1, temp + 5);
    return temp[3];
}

i와 j의 자리를 1로 작성하여 왼쪽 상단 박스에서만 값을 가져오게 했었다.

그러다 보니 제일 작은 예시 

2
3 4
2 1

로 주었을 때 알아차리지 못했다.


[AC 코드]

#include <iostream>
#include <algorithm>
using namespace std;

int n;
int arr[1025][1025];

int dq(int num, int i, int j){
    if(num == 1){
        return arr[i][j];
    }
    
    int temp[5];
    temp[1] = dq(num / 2, i, j);
    temp[2] = dq(num / 2, i + num / 2, j);
    temp[3] = dq(num / 2, i, j + num / 2);
    temp[4] = dq(num / 2, i + num / 2, j + num / 2);
    
    sort(temp + 1, temp + 5);
    return temp[3];
}

int main(){
    cin >> n;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= n; j++){
            cin >> arr[i][j];
        }
    }
    
    cout << dq(n, 1, 1);
    return 0;
}
728x90