统计子矩阵

题目描述 

给定一个 N × M 的矩阵A,请你统计有多少个子矩阵(最小 1 × 1,最大 N × M) 满足：
子矩阵中所有数的和不超过给定的整数K?

输入格式 

第一行包含三个整数N, M 和K.
之后 N 行每行包含 M 个整数，代表矩阵A.
30%的测试数据：1≤N,M≤20；
70%的测试数据：1≤N,M≤100；
100%的测试数据：1≤N,M≤500；0≤Aij≤1000；1≤K≤250000000。

输出格式 

一个整数代表答案。

输入样例

3 4 10
1 2 3 4
5 6 7 8
9 10 11 12

输出样例 

 19

代码 

#include<stdio.h>
int a[1010][1010],b[1010][1010];
int main()
{
	
	long long m,n,k;
	scanf("%lld%lld%lld\n",&m,&n,&k);
			
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
		{	scanf("%d",&a[i][j]);
			b[i][j]=a[i][j]+b[i][j-1]+b[i-1][j]-b[i-1][j-1];
		}
			
	}
	int t=0;
	for(int p=1;p<=m;p++){
		for(int q=1;q<=n;q++){
			for(int i=p;i<=m;i++){
				for(int j=q;j<=n;j++){
					int sum=b[i][j]-b[i][q-1]-b[p-1][j]+b[p-1][q-1];
					if(sum<=k)
						t++;
		}			
}
	}	
	}
	printf("%d",t);
	return 0;
 } 

连用四个for语句时间复杂度高,会导致时间超限,因此用p,q双指针来解决问题

代码 

#include <bits/stdc++.h>

using namespace std;

int n, m, k;
int a[505][505];
long long sum = 0;

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]+a[i][j - 1]-a[i - 1][j - 1];
        }//求出矩阵的前缀和
    }
    for (int i = 1; i <= m; ++i) {
        for (int j = i; j <= m; ++j) {//子矩阵的列
            for (int p = 1, q = 1; q <= n; ++q) {//子矩阵的行
                while (p <= q && a[q][j] - a[q][i - 1] - a[p - 1][j] + a[p - 1][i - 1] > k) {
                    p++;
                }
                if (p <= q) {//行每趟符合条件求和
                    sum += (q - p + 1);
                }
            }
        }
    }
    cout <<sum;
    return 0;
}

C语言中的printf和C++中的cout效率问题:printf比cout快(将近三倍)

(1)printf只是一个函数,cout是ostream的一个对象,而且还可以重载,

(2)比如输出一个结构体的内容

    printf只能将结构体的成员一个一个输出
    cout经过重载后却可以将结构体的全部成员输出

(3)cout语句的一般格式为：
    cout<<表达式1<<表达式2<<……<<表达式n;
    cin语句的一般格式为：
    cin>>变量1>>变量2>>……>>变量n;

    不能用cin语句把空格字符和回车换行符作为字符输入给字符变量,它们将被跳过｡如果想将      空格字符或回车换行符(或任何其他键盘上的字符)输入给字符变量,可以用getchar函数｡

C++的头文件声明

#include<bits/stdc++.h>//C++中的万能头文件,几乎包含了C++中所有的头文件,在大部分情况下使用这一行就可以省去其他库的生声明
#include <iostream>      //引用C++标准头文件库
using namespace std;      //使用名字空间

一些在C语言中常用的头文件在C++中的名字变为去掉“ .h”,并在开头增加字符c  

#include <cstring>

而在C语言中引用C++的头文件时

#include <iostream.h>    //引用头文件,不需要声明名字空间