Codeforces（D. Rarity and New Dress）DP

这场开局半小时没读懂A题意，还闹肚子了(o(╥﹏╥)o)，本来想把这场鸽了的，奈何上厕所的时候出了B题思路，然后又打下去了。rank+29，又是在1700左右徘徊，关注列表又一个好友上紫了，我啥时候也能上紫啊啊啊啊(o(╥﹏╥)o)
D题暴力写了一发WA5（连TLE的样例都没到就给我WA了），DP题做的还是太少了啊，得去搞个DP专题练练了

题意：
就是找给定区域里有多少个斜45度正方形，正方形内的字母必须都一样
枚举每个点，f[i][j]表示以该点为（正方形最低端的点）的正方形的边长最大是多少，z[i][j]表示左斜上方有多少连续相同的字母个数，y[i][j]表示右斜上方有多少连续相同的字母个数

代码：

#include 
#define ll long long
#define IOS std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
template<class T> inline void read(T &x){
    x=0; register char c=getchar(); register bool f=0;
    while(!isdigit(c))f^=c=='-',c=getchar();
    while(isdigit(c))x=x*10+c-'0',c=getchar(); if(f)x=-x;
}
using namespace std;

string s[2005];
int a[2005][2005],f[2005][2005];
int z[2005][2005],y[2005][2000];
int main()
{
    int n,m;
    cin>>n>>m;
    ll ans=n*m;
    for(int i=0;i<n;i++)
    {cin>>s[i];}
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {a[i+1][j+1]=(s[i][j]-'a')+1;}
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(a[i-1][j-1]==a[i][j])
            {z[i][j]=z[i-1][j-1]+1;}
            if(a[i-1][j+1]==a[i][j])
            {y[i][j]=y[i-1][j+1]+1;}
        }
    }
    int u,v;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(a[i-1][j]==a[i][j])
            {
                u=f[i-1][j]+1;
                if(z[i][j]>=u&&y[i][j]>=u&&y[i-u][j-u]>=u&&z[i-u][j+u]>=u)
                {f[i][j]=u;continue;}
                v=min(z[i][j],y[i][j]);
                if(v>=u){v=u-1;}
                f[i][j]=v;
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {ans+=f[i][j];}
    }
    cout<<ans<<endl;
    return 0;
}