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;
}