BOJ)2578 빙고

문제 : https://www.acmicpc.net/problem/2578

25개의 칸을 5X5 이중배열로 생각하였다가, 시간이 너무 오래 걸릴 것 같아 1차원배열의 연장으로 생각하였다.

#include <iostream> #include <vector> using namespace std; int map[25]; int cnt=0; int search(int n) { for (int i = 0; i < 25; i++) { if (map[i] == n) { map[i] = 0; return i; } } return -1; } void check_line(int position) { int sum = 0; for (int i = 0; i < 5; i++) { sum += map[(position / 5) * 5 + i]; if (sum > 0) break; } if (sum == 0) cnt++; sum = 0; for (int i = 0; i < 5; i++) { sum += map[(position % 5) + i * 5]; if (sum > 0) break; } if (sum == 0) cnt++; if (position != 24 && position % 4 == 0) { sum = 0; for (int i = 0; i < 5; i++) { sum += map[4 * (i + 1)]; if (sum > 0) break; } if (sum == 0) cnt++; } if (position % 6 == 0) { sum = 0; for (int i = 0; i <= 5; i++) { sum += map[6 * (i + 1)]; if (sum > 0) break; } if (sum == 0) cnt++; } } int main() { for (int i = 0; i < 25; i++) { cin >> map[i]; } int input; vector <int> res; for (int i = 0; i < 25; i++) { cin >> input; int position = search(input); check_line(position); if (cnt >= 3) { res.push_back(i+1); } } cout << res[0] << endl; }