제출 #1138359

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1138359		mychecksedad	Wombats (IOI13_wombats)	C++20	100 / 100	2797 ms	61836 KiB

wombats

#include "wombats.h"
#include<bits/stdc++.h>
using namespace std;
#define ll long long int
#define MOD (1000000000+7)
#define MOD1 (998244353)
#define pb push_back
#define all(x) x.begin(), x.end()
#define en cout << '\n'
#define ff first
#define ss second
#define pii pair<int,int>
#define vi vector<int>
const int C = 200, K = 50;


int T[(5000/K)*4][C][C], H[5000][200], V[5000][200], pref[5000][200], nxt[C][C], opt[C][C];
int get_pref(int row, int l, int r){
  if(r < l) swap(l, r);
  return pref[row][r] - pref[row][l];
}

int n, m;

void calc(int l, int r, int k){
  for(int j = 0; j < m; ++j){
    for(int i = 0; i < m; ++i) T[k][j][i] = get_pref(l, j, i), nxt[i][j] = MOD;
  }
  for(int row = l + 1; row <= r; ++row){
    for(int s = 0; s < m; ++s){
      for(int e = 0; e < m; ++e){
        nxt[s][e] = min(e == 0 ? MOD : nxt[s][e - 1] + H[row][e - 1], T[k][s][e] + V[row - 1][e]);
      }
    }
    for(int s = 0; s < m; ++s){
      for(int e = m - 1; e >= 0; --e){
        nxt[s][e] = min(e + 1 == m ? MOD : nxt[s][e + 1] + H[row][e], nxt[s][e]);
        T[k][s][e] = nxt[s][e];
      }
    }
  }
}

void comb(int k, int mid){
  for(int dif = m - 1; dif >= -(m-1); --dif){
    for(int s = 0; s < m; ++s){
      int e = s - dif;
      if(e >= 0 && e < m){
        int lb = e == 0 ? 0 : opt[s][e - 1];
        int rb = s == m - 1 ? m - 1 : opt[s + 1][e];
        T[k][s][e] = MOD;
        opt[s][e] = lb;
        for(int j = lb; j <= rb; ++j){
          int val = T[k<<1][s][j] + T[k<<1|1][j][e] + V[mid][j];
          if(T[k][s][e] > val){
            opt[s][e] = j;
            T[k][s][e] = val;
          }
        }
      }
    }
  }
}

void build(int l, int r, int k){
  if(r - l <= K){
    calc(l, r, k);
    return;
  }
  int mid = l+r>>1;
  build(l, mid, k<<1);
  build(mid+1, r, k<<1|1);
  comb(k, mid);
}

void upd(int l, int r, int p, int k){
  if(r - l <= K){
    calc(l, r, k);
    return;
  }
  int mid = l+r>>1;
  if(p <= mid) upd(l, mid, p, k<<1);
  else upd(mid+1, r, p, k<<1|1);
  comb(k, mid);
}

void init(int R, int C, int HH[5000][200], int VV[5000][200]) {
  n = R;
  m = C;
  for(int i = 0; i < n; ++i){
    for(int j = 0 ;j < m; ++j){
      H[i][j] = HH[i][j];
      V[i][j] = VV[i][j];
      if(j == 0) pref[i][0] = 0;
      else pref[i][j] = pref[i][j - 1] + H[i][j - 1];
    }
  }
  build(0, n - 1, 1);
}

void changeH(int P, int Q, int W) {
  H[P][Q] = W;
  for(int j = 0 ;j < m; ++j){
    if(j == 0) pref[P][0] = 0;
    else pref[P][j] = pref[P][j - 1] + H[P][j - 1];
  }
  upd(0, n - 1, P, 1);
}

void changeV(int P, int Q, int W) {
  V[P][Q] = W;
  upd(0, n - 1, P, 1);
}

int escape(int V1, int V2) {
  return T[1][V1][V2];
}

• Subtask 1: 9 / 9
• Subtask 2: 12 / 12
• Subtask 3: 16 / 16
• Subtask 4: 18 / 18
• Subtask 5: 21 / 21
• Subtask 6: 24 / 24