結果

問題 No.866 レベルKの正方形
ユーザー tonegawa
提出日時 2020-08-20 15:01:55
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,800 bytes
コンパイル時間 1,094 ms
コンパイル使用メモリ 94,464 KB
最終ジャッジ日時 2025-01-13 04:25:34
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main(int, const char**)’:
main.cpp:18:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   18 |   scanf("%d %d %d", &H, &W, &K);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <deque>
#include <queue>
#include <algorithm>
#include <set>
#include <map>
#define vv(a, b, c, d) vector<vector<d> >(a, vector<d>(b, c))
#define vvi vector<vector<int> >
#define vvl vector<vector<ll> >
#define MODs 1000000007;
typedef long long int ll;
using namespace std;

int main(int argc, char const *argv[]) {
  int H, W, K;
  scanf("%d %d %d", &H, &W, &K);
  std::vector<string> c(H);
  for(int i=0;i<H;i++) std::cin >> c[i];
  vvi dp = vv(H, 26*W, 0, int);
  for(int i=0;i<26;i++){
    dp[H-1][W-1+i*W] = (c[H-1][W-1]-'a'==i?1:-1);
    for(int j=0;j<H;j++){
      for(int k=0;k<W;k++){
        if(j==0&&k==0) continue;
        if(c[H-1-j][W-1-k]-'a'==i) dp[H-1-j][W-1-k+i*W] = 1;
        else if(j==0) dp[H-1-j][W-1-k + i*W] = (dp[H-1-j][W-k+i*W]==-1?-1:dp[H-1-j][W-k+i*W]+1);
        else if(k==0) dp[H-1-j][W-1-k+i*W] = (dp[H-j][W-1-k+i*W]==-1?-1:dp[H-j][W-1-k+i*W]+1);
        else {
          if(max({dp[H-j][W-1-k+i*W], dp[H-1-j][W-k+i*W], dp[H-j][W-k+i*W]})==-1) dp[H-1-j][W-1-k+i*W]=-1;
          else dp[H-1-j][W-1-k+i*W] = min({(dp[H-j][W-1-k+i*W]==-1?1000000007:dp[H-j][W-1-k+i*W]+1),
             (dp[H-1-j][W-k+i*W]==-1?1000000007:dp[H-1-j][W-k+i*W]+1), (dp[H-j][W-k+i*W]==-1?1000000007:dp[H-j][W-k+i*W]+1)});
        }
        if(dp[H-1-j][W-1-k+i*W]!=-1&&(j+1<dp[H-1-j][W-1-k+i*W]||k+1<dp[H-1-j][W-1-k+i*W])) dp[H-1-j][W-1-k+i*W]=-1;
      }
    }
  }
  ll ans = 0;
  for(int i=0;i<H;i++){
    for(int j=0;j<W;j++){
      std::vector<int> las(26);
      for(int k=0;k<26;k++) las[k] = (dp[i][j+k*W]==-1?1000000007:dp[i][j+k*W]);
      sort(las.begin(), las.end());
      if(las[K-1]==1000000007) continue;
      ans += min({H-i + 1, W-j+ 1, las[K]}) - las[K-1] ;
    }
  }
  printf("%lld\n", ans);
  return 0;
}
0