제출 #1273127

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1273127		strange420	세계 지도 (IOI25_worldmap)	C++20	15 / 100	19 ms	3108 KiB

worldmap

#include "worldmap.h"
#include <algorithm>
#include <iostream>
#include <functional>
#include <utility>
#include <cmath>
#include <cassert>

std::vector<int> spanning_tree(std::vector<std::vector<int>>& adj_list, std::vector<std::vector<int>>& adj_matrix) {
  std::function<bool(int, int)> dfs;
  std::vector<int> visited(adj_list.size(), 0), path;
  dfs = [&](int u, int p) {
    if (visited[u]) return false;
    visited[u] = 1;
    path.push_back(u);
    for (int v: adj_list[u])
      if (dfs(v, u)) path.push_back(u);
    return true;
  }; dfs(1, 0);

  for (int i=1; i<path.size(); i++) 
    adj_matrix[path[i-1]][path[i]] = adj_matrix[path[i]][path[i-1]] = 0;
  
  std::vector<std::vector<int>> new_adj_list(adj_list.size());
  for (int i=1; i<adj_matrix.size(); i++)
    for (int j=1; j<adj_matrix.size(); j++)
      if (adj_matrix[i][j])
        new_adj_list[i].push_back(j);
  adj_list = new_adj_list;
  return path;
}

std::vector<std::vector<int>> create_map(int N, int M, std::vector<int> A, std::vector<int> B) {
  // if (M == 0) {
  //   for (int i=1; i<=N; i++) {
  //     for (int j=i+1; j<=N; j++) {
  //       A.push_back(i);
  //       B.push_back(j);
  //     }
  //   }

  //   M = A.size();
  // }
  std::vector<std::vector<int>> adj_list(N+1);
  std::vector<std::vector<int>> adj_matrix(adj_list.size(), std::vector<int>(adj_list.size(), 0));
  for (int i=0; i<M; i++) {
    adj_list[A[i]].push_back(B[i]);
    adj_list[B[i]].push_back(A[i]);
    adj_matrix[A[i]][B[i]] = adj_matrix[B[i]][A[i]] = 1;
  }
  std::vector<int> partial = spanning_tree(adj_list, adj_matrix);

  // Diagonalization - Insert a new edge and take advantage of that edge
  int K = partial.size();
  std::vector<std::vector<int>> ans(K, std::vector<int>(K, 0));
  for (int i=0; i<K; i++) ans[0][i] = partial[i];

  for (int i=1; i<K; i++) {
    // Merge node
    int index = -1;
    for (int j=1; j<K-1 && index == -1; j++) {
      if (ans[i-1][j-1] == ans[i-1][j+1]) {
        ans[i][j] = ans[i-1][j-1];
        index = j;
      }
    }

    for (int j=index+1; j<K; j++) {
      for (int x=N; x>=1; x--) {
        if (adj_matrix[ans[i-1][j]][x] && adj_matrix[x][ans[i][j-1]]) {
          ans[i][j] = x;
          adj_matrix[ans[i-1][j]][x] = 0;
          adj_matrix[x][ans[i-1][j]] = 0;
          adj_matrix[ans[i][j-1]][x] = 0;
          adj_matrix[x][ans[i][j-1]] = 0;
          break;
        }
      }

      if (ans[i][j] == 0) {
        ans[i][j] = ans[i-1][std::min(j+1, K-1)];
      }
    }

    for (int j=index-1; j>=0; j--) {
      for (int x=1; x<=N; x++) {
        if (adj_matrix[ans[i-1][j]][x] && adj_matrix[x][ans[i][j+1]]) {
          ans[i][j] = x;
          adj_matrix[ans[i-1][j]][x] = 0;
          adj_matrix[x][ans[i-1][j]] = 0;
          adj_matrix[ans[i][j+1]][x] = 0;
          adj_matrix[x][ans[i][j+1]] = 0;
          break;
        }
      }

      if (ans[i][j] == 0) {
        ans[i][j] = ans[i-1][std::max(j-1, 0)];
      }
    }
  }

  // Write a checker to debug.
  int validate[N+5][N+5];
  for (int i=0; i<K; i++) {
    for (int j=0; j<K; j++) {
      if (i-1 >= 0) {
        validate[ans[i][j]][ans[i-1][j]] = 1;
        validate[ans[i-1][j]][ans[i][j]] = 1;
      }

      if (j-1 >= 0) {
        validate[ans[i][j]][ans[i][j-1]] = 1;
        validate[ans[i][j-1]][ans[i][j]] = 1;
      }
    }
  }

  // std::cout << "Final size: " << ans.size() << '\t' << M << '\n';
  // for (int i=0; i<ans.size(); i++) {
  //   for (int j=0; j<ans.size(); j++) {
  //     std::cout << ans[i][j] << '\t';
  //   }
  //   std::cout << '\n';
  // }
  // for (int i=1; i<=N; i++)
  //   for (int j=i+1; j<=N; j++)
  //     if (validate[i][j] == 0) {
  //       std::cout << i << '\t' << j << '\n';
  //       exit(-1);
  //     }
  return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 5 / 5
• Subtask 2: 10 / 10
• Subtask 3: 0 / 7
• Subtask 4: 0 / 8
• Subtask 5: 0 / 14
• Subtask 6: 0 / 56