제출 #316888

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
316888		spatarel	식물 비교 (IOI20_plants)	C++17	0 / 100	4059 ms	3108 KiB

plants

#include "plants.h"
#include <limits.h>
#include <stdio.h>

struct MoxRange {
  int mox;
  int l;
  int r;
};

MoxRange getMoxRange(int k, std::vector<int> &r) {
  int n = r.size();
  int i = n - 1;
  int positives = 0;
  while (i >= 0 && r[i] != 0) {
    i--;
    positives++;
  }
  int maxPositives = -1;
  int maxZero = 0;
  for (int i = 0; i < n; i++) {
    if (r[i] == 0 && maxPositives < positives) {
      maxPositives = positives;
      maxZero = i;
    }
    if (r[i] != 0) {
      positives++;
    } else {
      positives = 0;
    }
    //printf("i=%d posi=%d\n", i, positives);
  }
  MoxRange answer;
  answer.mox = maxZero;
  answer.l = maxZero - maxPositives;
  answer.r = maxZero + k - 1;
  /*
  if (answer.r - answer.l + 1 > n) {
    answer.r = answer.l + n - 1;
  }
  answer.l = (answer.l + n) % n;
  answer.r = (answer.r + n) % n;//*/
  
  r[maxZero] = -1;
  i = maxZero;
  for (int i = 0; i < k - 1; i++) {
    maxZero = (maxZero - 1 + n) % n;
    r[maxZero]--;
  }
	return answer;
}

void print(std::vector<int> &r) {
  int n = r.size();
  for (int i = 0; i < n; i++) {
    printf("%d ", r[i]);
  }
  printf("\n");
}

const int MAX_N = 5000;

int n;
int gt[MAX_N][MAX_N];
int leftGt[MAX_N];
int rightGt[MAX_N];

void init(int k, std::vector<int> r) {
  n = r.size();
  /*
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < n; j++) {
      gt[i][j] = 0;
    }
  }//*/
  for (int i = 0; i < n; i++) {
    leftGt[i] = INT_MIN;
    rightGt[i] = INT_MAX;
  }
  for (int i = 0; i < n; i++) {
    //print(r);
    MoxRange moxRange = getMoxRange(k, r);
    //printf("%d > [%d..%d]\n", moxRange.mox, moxRange.l, moxRange.r);
    
    for (int i = moxRange.mox + 1; i <= moxRange.r; i++) {
      if (r[i % n] >= 0) {
        if (i < n && leftGt[i] < moxRange.mox) {
          leftGt[i] = moxRange.mox;
        }
        if (i >= n && leftGt[i - n] < moxRange.mox - n) {
          leftGt[i - n] = moxRange.mox - n;
        }
      }
    }
    for (int i = moxRange.mox - 1; i >= moxRange.l; i--) {
      if (r[(i + n) % n] >= 0) {
        if (0 <= i && moxRange.mox < rightGt[i]) {
          rightGt[i] = moxRange.mox;
        }
        if (0 > i && moxRange.mox + n < rightGt[i + n]) {
          rightGt[i + n] = moxRange.mox + n;
        }
      }
    }
    
    /*
    int x = moxRange.l;
    do {
      if (x != moxRange.mox && gt[x][moxRange.mox] == 0 && r[x] >= 0) {
        gt[moxRange.mox][x] = +1;
        gt[x][moxRange.mox] = -1;
      }
      x = (x + 1) % n;
    } while (x != (moxRange.r + 1) % n);//*/
  }
  /*
  for (int k = 0; k < n; k++) {
    for (int i = 0; i < n; i++) {
      for (int j = 0; j < n; j++) {
        for (int value : {-1, +1}) {
          if (gt[i][k] == value && gt[k][j] == value) {
            gt[i][j] = value;
          }
        }
      }
    }
  }//*/
  /*
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < n; j++) {
      printf("%d ", gt[i][j]);
    }
    printf("\n");
  }//*/
  
  /*
  for (int i = 0; i < n; i++) {
    printf("%d ", rightGt[i]);
  }
  printf("\n");
  for (int i = 0; i < n; i++) {
    printf("%d ", leftGt[i]);
  }
  printf("\n");//*/
  return;
}

int compare_plants(int x, int y) {
  //assert(x < y);
  
  int src[] = { x,  y};
  int dest[]= { y,  x};
  int ans[] = {-1, +1};
  for (int i : {0, 1}) {
    int pos = src[i];
    while (rightGt[pos] != INT_MAX && pos != dest[i]) {
      pos = rightGt[pos] % n;
    }
    if (pos == dest[i]) {
      return ans[i];
    }
    pos = y;
    while (leftGt[pos] != INT_MIN && pos != x) {
      pos = leftGt[pos] % n;
    }
    if (pos == x) {
      return +1;
    }
  }
  //return gt[x][y];
  return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 14
• Subtask 3: 0 / 13
• Subtask 4: 0 / 17
• Subtask 5: 0 / 11
• Subtask 6: 0 / 15
• Subtask 7: 0 / 25