結果

問題 No.655 E869120 and Good Triangles
コンテスト
ユーザー hos.lyric
提出日時 2019-07-21 17:59:34
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 1,413 ms / 2,500 ms
コード長 3,664 bytes
コンパイル時間 979 ms
コンパイル使用メモリ 121,272 KB
実行使用メモリ 325,488 KB
最終ジャッジ日時 2024-06-22 01:55:00
合計ジャッジ時間 27,277 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.functional, std.range, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.complex, std.container, std.math, std.numeric, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) { throw new EOFException; } tokens = readln.split; } auto token = tokens.front; tokens.popFront; return token; }
int readInt() { return readToken.to!int; }
long readLong() { return readToken.to!long; }
real readReal() { return readToken.to!real; }

bool chmin(T)(ref T t, in T f) { if (t > f) { t = f; return true; } else { return false; } }
bool chmax(T)(ref T t, in T f) { if (t < f) { t = f; return true; } else { return false; } }

int binarySearch(alias pred, T)(in T[] as) { int lo = -1, hi = cast(int)(as.length); for (; lo + 1 < hi; ) { const mid = (lo + hi) >> 1; (unaryFun!pred(as[mid]) ? hi : lo) = mid; } return hi; }
int lowerBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a >= val)); }
int upperBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a > val)); }


/*
        00      
      10  01    
    20  11  02  
  30  21  12  03
*/

int N, K;
long P;
int[] X, Y;

int[][] D;

void bfs() {
  auto q = new int[N * (N + 1)];
  int qHead, qTail;
  D = new int[][](N, N);
  foreach (x; 0 .. N) foreach (y; 0 .. N) {
    if (x + y < N) {
      D[x][y] = 2 * N;
    }
  }
  foreach (k; 0 .. K) {
    D[X[k]][Y[k]] = 0;
    q[qTail++] = X[k];
    q[qTail++] = Y[k];
  }
  for (; qHead != qTail; ) {
    const x = q[qHead++];
    const y = q[qHead++];
    void update(int xx, int yy) {
      if (D[xx][yy] > D[x][y] + 1) {
        D[xx][yy] = D[x][y] + 1;
        q[qTail++] = xx;
        q[qTail++] = yy;
      }
    }
    if (x + y + 1 < N) {
      update(x + 1, y);
      update(x, y + 1);
    }
    if (x - 1 >= 0) {
      update(x - 1, y + 1);
      update(x - 1, y);
    }
    if (y - 1 >= 0) {
      update(x, y - 1);
      update(x + 1, y - 1);
    }
  }
}

void main() {
  try {
    for (; ; ) {
      N = readInt();
      K = readInt();
      P = readLong();
      X = new int[K];
      Y = new int[K];
      foreach (k; 0 .. K) {
        X[k] = readInt() - 1;
        Y[k] = readInt() - 1;
        X[k] -= Y[k];
      }
      
      bfs();
      
      // D[x][0] + D[x][1] + ...
      auto sumX = new long[][](N, N + 1);
      foreach (x; 0 .. N) {
        foreach (y; 0 .. N - x) {
          sumX[x][y + 1] = sumX[x][y] + D[x][y];
        }
      }
      // D[0][y] + D[1][y] + ...
      auto sumY = new long[][](N, N + 1);
      foreach (y; 0 .. N) {
        foreach (x; 0 .. N - y) {
          sumY[y][x + 1] = sumY[y][x] + D[x][y];
        }
      }
      
      debug {
        foreach (z; 0 .. N) {
          write(' '.repeat(N - z));
          foreach (y; 0 .. z + 1) {
            write(D[z - y][y], " ");
          }
          writeln();
        }
        writeln("sumX = ", sumX);
        writeln("sumY = ", sumY);
      }
      
      long ans;
      foreach (z; 0 .. N) {
        long now;
        for (int y0 = 0, y1 = 0; y0 <= z; ++y0) {
          // D[z - y1][y0] + ... + D[z - y1][y1]
          for (; y1 <= z && (y1 < y0 || now + (sumX[z - y1][y1 + 1] - sumX[z - y1][y0]) < P); ++y1) {
            now += (sumX[z - y1][y1 + 1] - sumX[z - y1][y0]);
          }
          debug {
            writeln(z, " ", y0, " ", y1);
          }
          ans += z + 1 - y1;
          // D[z - (y1 - 1)][y0] + ... + D[z - y0][y0]
          now -= (sumY[y0][z - y0 + 1] - sumY[y0][z - (y1 - 1)]);
        }
      }
      writeln(ans);
    }
  } catch (EOFException e) {
  }
}
0