結果

問題 No.866 レベルKの正方形
コンテスト
ユーザー LayCurse
提出日時 2019-08-17 01:16:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,573 bytes
コンパイル時間 2,476 ms
コンパイル使用メモリ 196,208 KB
最終ジャッジ日時 2025-01-07 13:11:40
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 2 WA * 20
権限があれば一括ダウンロードができます

ソースコード

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[2001][2001][26];
inline int getSum(int k, int i, int j, int v){
  if(g[i+v][j+v][k] - g[i][j+v][k] - g[i+v][j][k] + g[i][j][k]){
    return 1;
  }
  return 0;
}
inline int 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(i=0;i<(X);i++){
    for(j=0;j<(Y);j++){
      for(k=0;k<(26);k++){
        g[i+1][j+1][k] = g[i+1][j][k] + g[i][j+1][k] - g[i][j][k];
      }
      g[C[i][j]][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[2001][2001][26];
// 
// inline int getSum(int k, int i, int j, int v){
//   if(g[i+v][j+v][k] - g[i][j+v][k] - g[i+v][j][k] + g[i][j][k]) return 1;
//   return 0;
// }
// 
// inline int 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(i,X) rep(j,Y){
//     rep(k,26) g[i+1][j+1][k] = g[i+1][j][k] + g[i][j+1][k] - g[i][j][k];
//     g[C[i][j]][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