#include <bits/stdc++.h>
#include "worldmap.h"
using namespace std;
std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
const int K = max(7 * N, 240); // đảm bảo phá subtask 6
vector<vector<int>> ans(K, vector<int>(K, 1)); // mặc định toàn quốc gia 1
// Đảm bảo mỗi quốc gia có mặt ít nhất một lần
for (int i = 0; i < N; ++i) {
int row = (i * 11) % K;
int col = (i * 17) % K;
ans[row][col] = i + 1;
}
// Đặt các cặp A[i] và B[i] vào các ô liền kề
int row = 0, col = 0;
for (int i = 0; i < M; ++i) {
// Đặt A[i] và B[i] cạnh nhau, ngang hoặc dọc
ans[row][col] = A[i];
ans[row][col + 1] = B[i];
row += 2;
if (row + 1 >= K) {
row = 0;
col += 3;
if (col + 1 >= K) break; // tránh vượt lưới
}
}
return ans;
}
