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 <bits/stdc++.h>
#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;
}

猜你喜欢

转载自blog.csdn.net/qq_43781431/article/details/107883696