https://zerojudge.tw/ShowProblem?problemid=g501

#include <bits/stdc++.h> using namespace std; int main(){ cin.tie(0); ios_base::sync_with_stdio(false); vector<int> v = { {1, 2, 3, 5, 6, 8, 11, 13, 17, 21, 22}, {2, 3, 5, 7, 8, 13, 16, 17, 21, 22, 24}, {1, 3, 5, 6, 7, 15, 16, 17, 22, 23, 24, 25}, {2, 4, 5, 7, 8, 15, 17, 21, 23, 25} } int arr[25][4]; for(int i=0; i<25; i++){ for(int j=0; j<4; j++){ arr[ v[i][j] ] = 0; } } for(int i=0; i<4; i++){ for(int j=0; j<v[i].size(); j++){ arr[ v[i][j]-1 ]++; } } int h, n; cin>>h>>n; h -= 1; }