結果

問題 No.1916 Making Palindrome on Gird
ユーザー umezoumezo
提出日時 2022-04-29 23:49:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 4,015 bytes
コンパイル時間 3,023 ms
コンパイル使用メモリ 227,680 KB
最終ジャッジ日時 2025-01-29 00:37:04
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 MLE * 2
other AC * 20 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

const int MOD=1e9+7;
map<string,ll> dp1[205][205],dp2[205][205];

int main(){
  int h,w;
  cin>>h>>w;
  vector<string> S(h);
  rep(i,h) cin>>S[i];
 
  if(h==1 && w==1){
    cout<<1<<endl;
    return 0;
  }
  
  string ss1,ss2;
  ss1+=S[0][0],ss2+=S[h-1][w-1];
  if(ss1!=ss2){
    cout<<0<<endl;
    return 0;
  }
  dp1[0][0][ss1]=1,dp2[0][0][ss2]=1;
  int s=h+w-2;
  if(s%2==1){
    s=(s-1)/2;
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        if(i<h-1 && j<w-1){
          for(auto [a,b]:dp1[i][j]){
            dp1[i+1][j][a+S[i+1][j]]=(dp1[i+1][j][a+S[i+1][j]]+b)%MOD;
            dp1[i][j+1][a+S[i][j+1]]=(dp1[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
        else if(i<h-1){
          for(auto [a,b]:dp1[i][j]){
            dp1[i+1][j][a+S[i+1][j]]=(dp1[i+1][j][a+S[i+1][j]]+b)%MOD;
          }
        }
        else{
          for(auto [a,b]:dp1[i][j]){
            dp1[i][j+1][a+S[i][j+1]]=(dp1[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
      }
    }
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        swap(S[i][j],S[h-1-i][w-1-j]);
      }
    }
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        if(i<h-1 && j<w-1){
          for(auto [a,b]:dp2[i][j]){
            dp2[i+1][j][a+S[i+1][j]]=(dp2[i+1][j][a+S[i+1][j]]+b)%MOD;
            dp2[i][j+1][a+S[i][j+1]]=(dp2[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
        else if(i<h-1){
          for(auto [a,b]:dp2[i][j]){
            dp2[i+1][j][a+S[i+1][j]]=(dp2[i+1][j][a+S[i+1][j]]+b)%MOD;
          }
        }
        else{
          for(auto [a,b]:dp2[i][j]){
            dp2[i][j+1][a+S[i][j+1]]=(dp2[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
      }
    }
    ll ans=0;
    rep(i,h){
      int j=s-i;
      if(j<0 || j>=w) continue;
      map<string,ll> m2;
      if(h-i-2>=0) for(auto [a,b]:dp2[h-1-i-1][w-1-j]) m2[a]=(m2[a]+b)%MOD;
      if(w-j-2>=0) for(auto [a,b]:dp2[h-1-i][w-1-j-1]) m2[a]=(m2[a]+b)%MOD;
      for(auto [a,b]:dp1[i][j]) ans=(ans+b*m2[a]%MOD)%MOD;
    }
    cout<<ans<<endl;

  }
  else{
    s=s/2-1;
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        if(i<h-1 && j<w-1){
          for(auto [a,b]:dp1[i][j]){
            dp1[i+1][j][a+S[i+1][j]]=(dp1[i+1][j][a+S[i+1][j]]+b)%MOD;
            dp1[i][j+1][a+S[i][j+1]]=(dp1[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
        else if(i<h-1){
          for(auto [a,b]:dp1[i][j]){
            dp1[i+1][j][a+S[i+1][j]]=(dp1[i+1][j][a+S[i+1][j]]+b)%MOD;
          }
        }
        else{
          for(auto [a,b]:dp1[i][j]){
            dp1[i][j+1][a+S[i][j+1]]=(dp1[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
      }
    }
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        swap(S[i][j],S[h-1-i][w-1-j]);
      }
    }
    rep(i,h){
      rep(j,w){
        if(i+j>s) break;
        if(i<h-1 && j<w-1){
          for(auto [a,b]:dp2[i][j]){
            dp2[i+1][j][a+S[i+1][j]]=(dp2[i+1][j][a+S[i+1][j]]+b)%MOD;
            dp2[i][j+1][a+S[i][j+1]]=(dp2[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
        else if(i<h-1){
          for(auto [a,b]:dp2[i][j]){
            dp2[i+1][j][a+S[i+1][j]]=(dp2[i+1][j][a+S[i+1][j]]+b)%MOD;
          }
        }
        else{
          for(auto [a,b]:dp2[i][j]){
            dp2[i][j+1][a+S[i][j+1]]=(dp2[i][j+1][a+S[i][j+1]]+b)%MOD;
          }
        }
      }
    }
    s++;
    ll ans=0;
    rep(i,h){
      int j=s-i;
      if(j<0 || j>=w) continue;
      map<string,ll> m1,m2;
      if(i-1>=0) for(auto [a,b]:dp1[i-1][j]) m1[a]=(m1[a]+b)%MOD;
      if(j-1>=0) for(auto [a,b]:dp1[i][j-1]) m1[a]=(m1[a]+b)%MOD;
      if(h-i-2>=0) for(auto [a,b]:dp2[h-1-i-1][w-1-j]) m2[a]=(m2[a]+b)%MOD;
      if(w-j-2>=0) for(auto [a,b]:dp2[h-1-i][w-1-j-1]) m2[a]=(m2[a]+b)%MOD;
      for(auto [a,b]:m1) ans=(ans+b*m2[a]%MOD)%MOD;
    }
    cout<<ans<<endl;
  }
  
  return 0; 
}
0