# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1252543 | bzzzzzzzzzz | 세계 지도 (IOI25_worldmap) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
#include "worldmap.h"
using namespace std;
using ll = long long;
std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
int p = 0;
for (int i = 0; i < M; i++) {
if (A[i] == 1) {
p++;
}
if (B[i] == 1) {
p++;
}
}
if (M == N * (N - 1) / 2) {
vector<vector<int>> ans(240, vector<int> (240, 1));
int c = 0;
int h = 0;
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
ans[c][h] = i;
ans[c][h + 1] = j;
c++;
if (c == 240) {
c = 0;
h += 2;
}
}
}
return ans;
} else if (M == N - 1 && p != N - 1) {
vector<vector<int>> ans(N, vector<int> (N, 1));
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
ans[i][j] = i + 1;
}
}
return ans;
} else {
vector<vector<int>> ans(240, vector<int> (240, 1));
int c = 0;
int h = 0;
for (int i = 0; i < M; i++) {
ans[c][h] = A[i];
ans[c][h + 1] = B[j];
c++;
if (c == 240) {
c = 0;
h += 2;
}
}
return ans;
}
}