結果

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

ソースコード

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');
  }
}
template<class S, class T> inline S chmax(S &a, T b){
  if(a<b){
    a=b;
  }
  return a;
}
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, v1, v2;
  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[i+1][j+1][C[i][j]]++;
    }
  }
  for(ii=0;ii<(X);ii++){
    for(jj=0;jj<(Y);jj++){
      if(ii==0 || jj==0){
        i = ii;
        j = jj;
        v1 = v2 = 1;
        for(;;){
          while( i+v2+1 <= X && j+v2+1 <= Y && getSum(i,j,v2+1) <= K-1 ){
            v2++;
          }
          chmax(v1, v2);
          while( i+v1+1 <= X && j+v1+1 <= Y && getSum(i,j,v1+1) <= K   ){
            v1++;
          }
          if(K-1>=1){
            res += v1 - v2;
          }
          else{
            res += v1 ;
          }
          i++;
          j++;
          v1--;
          v2--;
          if(i==X || j==Y){
            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, v1, v2;
//   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[i+1][j+1][C[i][j]]++;
//   }
// 
//   rep(ii,X) rep(jj,Y) if(ii==0 || jj==0){
//     i = ii; j = jj; v1 = v2 = 1;
//     for(;;){
//       while( i+v2+1 <= X && j+v2+1 <= Y && getSum(i,j,v2+1) <= K-1 ) v2++;
//       v1 >?= v2;
//       while( i+v1+1 <= X && j+v1+1 <= Y && getSum(i,j,v1+1) <= K   ) v1++;
//       res += v1 if[K-1>=1, - v2];
//       i++; j++; v1--; v2--;
//       if(i==X || j==Y) break;
//     }
//   }
// 
//   wt(res);
// }
0