結果

問題 No.866 レベルKの正方形
ユーザー LayCurseLayCurse
提出日時 2019-08-16 22:19:29
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 3,624 bytes
コンパイル時間 2,403 ms
コンパイル使用メモリ 204,176 KB
実行使用メモリ 420,780 KB
最終ジャッジ日時 2024-09-22 17:52:48
合計ジャッジ時間 27,496 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,816 KB
testcase_01 AC 3 ms
6,940 KB
testcase_02 AC 4 ms
6,940 KB
testcase_03 AC 3 ms
6,944 KB
testcase_04 AC 3 ms
6,940 KB
testcase_05 AC 4 ms
6,944 KB
testcase_06 AC 3 ms
6,944 KB
testcase_07 AC 4 ms
6,944 KB
testcase_08 AC 4,526 ms
413,964 KB
testcase_09 AC 4,754 ms
413,964 KB
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 AC 5,406 ms
413,836 KB
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 AC 5,003 ms
413,968 KB
testcase_20 AC 4,357 ms
413,960 KB
testcase_21 AC 2,398 ms
413,960 KB
testcase_22 AC 3 ms
6,944 KB
testcase_23 AC 3 ms
6,944 KB
testcase_24 AC 5 ms
7,628 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
inline void rd(int &x){
  int k, m=0;
  x=0;
  for(;;){
    k = getchar_unlocked();
    if(k=='-'){
      m=1;
      break;
    }
    if('0'<=k&&k<='9'){
      x=k-'0';
      break;
    }
  }
  for(;;){
    k = getchar_unlocked();
    if(k<'0'||k>'9'){
      break;
    }
    x=x*10+k-'0';
  }
  if(m){
    x=-x;
  }
}
inline void rd(char &c){
  int i;
  for(;;){
    i = getchar_unlocked();
    if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF){
      break;
    }
  }
  c = i;
}
inline int rd(char c[]){
  int i, sz=0;
  for(;;){
    i = getchar_unlocked();
    if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF){
      break;
    }
  }
  c[sz++] = i;
  for(;;){
    i = getchar_unlocked();
    if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF){
      break;
    }
    c[sz++] = i;
  }
  c[sz]='\0';
  return sz;
}
inline void wt_L(char a){
  putchar_unlocked(a);
}
inline void wt_L(long long x){
  char f[20];
  int m=0, s=0;
  if(x<0){
    m=1;
    x=-x;
  }
  while(x){
    f[s++]=x%10;
    x/=10;
  }
  if(!s){
    f[s++]=0;
  }
  if(m){
    putchar_unlocked('-');
  }
  while(s--){
    putchar_unlocked(f[s]+'0');
  }
}
int X;
int Y;
int K;
char C[2000][2002];
int g[26][2001][2001];
inline long long getSum(int k, int i, int j, int v){
  if(g[k][i+v][j+v] - g[k][i][j+v] - g[k][i+v][j] + g[k][i][j]){
    return 1;
  }
  return 0;
}
inline long long getSum(int i, int j, int v){
  int k, res=0;
  for(k=0;k<(26);k++){
    res += getSum(k,i,j,v);
  }
  return res;
}
int main(){
  int i, ii, j, jj, k, loop, v;
  long long res=0;
  rd(X);
  rd(Y);
  rd(K);
  {
    int Lj4PdHRW;
    for(Lj4PdHRW=0;Lj4PdHRW<(X);Lj4PdHRW++){
      rd(C[Lj4PdHRW]);
    }
  }
  for(i=0;i<(X);i++){
    for(j=0;j<(Y);j++){
      C[i][j] -= 'a';
    }
  }
  for(k=0;k<(26);k++){
    for(i=0;i<(X);i++){
      for(j=0;j<(Y);j++){
        g[k][i+1][j+1] = g[k][i+1][j] + g[k][i][j+1] - g[k][i][j];
        if(C[i][j]==k){
          g[k][i+1][j+1]++;
        }
      }
    }
  }
  for(loop=0;loop<(2);loop++){
    for(ii=0;ii<(X);ii++){
      for(jj=0;jj<(Y);jj++){
        if(ii==0 || jj==0){
          i = ii;
          j = jj;
          v = 1;
          for(;;){
            while( i+v+1 <= X && j+v+1 <= Y && getSum(i,j,v+1) <= K ){
              v++;
            }
            res += (1 - 2*loop) * v;
            i++;
            j++;
            v--;
            if(i==X || j==Y){
              break;
            }
          }
        }
      }
    }
    K--;
    if(K==0){
      break;
    }
  }
  wt_L(res);
  wt_L('\n');
  return 0;
}
// cLay varsion 20190810-2

// --- original code ---
// int X, Y, K;
// char C[2000][2002];
// 
// int g[26][2001][2001];
// 
// inline ll getSum(int k, int i, int j, int v){
//   if(g[k][i+v][j+v] - g[k][i][j+v] - g[k][i+v][j] + g[k][i][j]) return 1;
//   return 0;
// }
// 
// inline ll getSum(int i, int j, int v){
//   int k, res = 0;
//   rep(k,26) res += getSum(k,i,j,v);
//   return res;
// }
// 
// {
//   int i, j, k;
//   int ii, jj, v;
//   ll res = 0;
//   
//   rd(X,Y,K,C(X));
// 
//   rep(i,X) rep(j,Y) C[i][j] -= 'a';
// 
//   rep(k,26) rep(i,X) rep(j,Y){
//     g[k][i+1][j+1] = g[k][i+1][j] + g[k][i][j+1] - g[k][i][j];
//     if(C[i][j]==k) g[k][i+1][j+1]++;
//   }
// 
//   rep(loop,2){
//     rep(ii,X) rep(jj,Y) if(ii==0 || jj==0){
//       i = ii; j = jj; v = 1;
//       for(;;){
//         while( i+v+1 <= X && j+v+1 <= Y && getSum(i,j,v+1) <= K ) v++;
//         res += (1 - 2loop) * v;
//         i++; j++; v--;
//         if(i==X || j==Y) break;
//       }
//     }
// 
//     K--;
//     if(K==0) break;
//   }
// 
//   wt(res);
// }
0