题目1:统计子矩阵
解题
前缀和+暴力枚举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:牛的学术圈
解题
本题也可以用二分+前缀和来做,这里只讲双指针的做法。
首先要明确,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);
}