D. Table with Letters - 2（双指针+前缀和）

思路

• 题意：给一个m*n的矩阵，求其中有多少个子矩阵满足a的个数小于等于K，并且四个顶点字符相同
• 做法：先用前缀和，记下a的个数，在枚举一下上边和下边，之后用类似于双指针的做法，我们令左边的纵坐标为p，右边纵坐标为q，先移动右边，直到右边点a的字符个数大于K，每次移动右边如果右边的两个顶点相同就使该点字符的个数加一，如果右边不能移动了，就看下当前和左边顶点字符一样的右边字符有几个，ans加上这个值，之后左边往右移动，别忘记把该字符出现的次数减一。

代码

#include<bits/stdc++.h>
const int maxn=1e5+10;
using namespace std;
int n,m,kk;
char a[500][500];
int sum[500][500];///sum[i][j]代表从(1,1)到第(i,j)位置的矩阵的前缀和
map<char,int> mp;///选取矩阵中各个字母的出现的对数
int main()
{
    scanf("%d%d%d",&n,&m,&kk);
    for(int i=1; i<=n; i++)
        scanf("%s",a[i]+1);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
        {
            sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
            if(a[i][j]=='a') sum[i][j]++;
        }
    long long ans=0;
    for(int i=1; i<=n; i++)
    {
        for(int j=i+1; j<=n; j++) ///枚举矩阵的上边i和下边j
        {
            int p=1;///代表矩阵右边
            mp.clear();
            for(int k=1; k<=m; k++) ///矩阵左边
            {
                if(a[i][k]!=a[j][k]) continue;///不满足矩形四个点相同
                while(p<=m&&sum[j][p]-sum[i-1][p]-sum[j][k-1]+sum[i-1][k-1]<=kk)
                {
                    if(a[i][p]==a[j][p]) mp[a[i][p]]++;
                    p++;///矩形满足a个数小于kk的要求，矩阵右边增大，直到不满足，那么增大左边k++
                }
                mp[a[i][k]]--;///减去重复加的它本身
                if(mp[a[i][k]]>0) ans+=mp[a[i][k]];///因为矩形左边固定，看右边满足四个点相同条件下有多少个这样的矩形
                //cout<<q[a[i][k]]<<endl;
            }
        }
    }
    cout << ans;
    return 0;
}