双指针的一些题目

题目1：统计子矩阵

4405. 统计子矩阵 - AcWing题库

解题

前缀和+暴力枚举x1、y1、x2、y2的做法是O()的，考虑用双指针进行优化。

优化的方式是：我们将每一列在[x1,x2]范围内的部分看作一个整体，用指针维护左右边界。

当当前矩阵的和大于k时，y1后移。

最后，我们为结果加上y2-y1+1（以y1开头，y2=[y1,y2]的子矩阵，共y2-y1+1个）。因为每次我们的y1、y2每次只向右移动一位，因此在计算时，只需要计算包含了y1这一列的子矩阵即可，如果计算全部的子矩阵，会在下一次计算时有重复。

#include<iostream>
using namespace std;
const int N=510;
int n,m,k;
int a[N][N];
int s[N][N];
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) 
        {
            scanf("%d",&a[i][j]);
            a[i][j]+=a[i-1][j];
        }
    long long res=0;
    for(int x1=1;x1<=n;x1++)
        for(int x2=x1;x2<=n;x2++)
            for(int y1=1,y2=1,sum=0;y2<=m;y2++)
            {
                sum+=(a[x2][y2]-a[x1-1][y2]);
                while(sum>k)
                {
                    sum-=(a[x2][y1]-a[x1-1][y1]);
                    y1++;
                }
                res+=(y2-y1+1);
            }
    printf("%lld",res);
}

题目2：牛的学术圈

3745. 牛的学术圈 I - AcWing题库

解题

本题也可以用二分+前缀和来做，这里只讲双指针的做法。

首先要明确，h必须满足两个条件：1.大于等于h的数不少于h个，2.等于h-1的数不超过l个。

第一个条件很好求，我们将c数组从大到小排列后，若c[h]>=h-1，则可判断正确。

对于第二个条件，双指针的做法是：设置指针j指向第一个>=h的数的下标，用i来枚举h，就会有c[i]>=i-1，c[j]>=i，而[j+1,i]区间内数的个数就是等于i-1(也就是h-1)的数的个数，为i-j。

ac代码如下：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int n,l;
int c[N];
bool cmp(int x,int y)
{
    return x>y;
}
int main()
{
    scanf("%d%d",&n,&l);
    for(int i=1;i<=n;i++) scanf("%d",&c[i]);
    sort(c+1,c+1+n,cmp);
    int res=0;
    for(int h=0,j=n;h<=n;h++)
    {
        while(j&&c[j]<h) j--;
        if(c[h]>=h-1&&h-j<=l) res=h;
    }
    printf("%d",res);
}