前缀和问题qcjj的优秀总结 传送门
1.Maximum sum—不限制长度的两段连续区间和最大值
题目链接
#include<iostream>
using namespace std;
const int N = 50010;
int f[N],g[N],F[N],G[N];//以i结尾的最大值
int w[N];
int n;
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
int maxv=-0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);
if(i==1)f[i]=w[i];
else f[i]=max(0,f[i-1])+w[i];
maxv=F[i]=max(maxv,f[i]);
}
maxv=-0x3f3f3f3f;
for(int i=n;i>=1;i--)
{
if(i==n)g[i]=w[n];
else g[i]=max(g[i+1],0)+w[i];
maxv=G[i]=max(g[i],maxv);
}
int res=-0x3f3f3f3f;
for(int i=1;i<n;i++)
{
res=max(res,F[i]+G[i+1]);
}
cout<<res<<endl;
}
}
2.数学考试–限制长度的两个连续区间和最大值
题目链接
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long
const int N = 200010 ;
int s[N];
int n,k;
template<class T>inline void read(T &res)
{
char c;T flag=1;
while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}
void solve()
{
read(n);
read(k);
for(int i=1;i<=n;i++)
{
read(s[i]);
s[i]+=s[i-1];
}
int maxv=-1e18;
int res=-1e18;
for(int i=k;i<=n-k;i++)
{
maxv=max(maxv,s[i]-s[i-k]);
res=max(res,maxv+s[i+k]-s[i]);
}
cout<<res<<endl;
}
signed main()
{
int t;
read(t);
while(t--)
{
solve();
}
}
3.To the Max
题目链接
暴力做法
预处理二维前缀和,枚举矩阵的左上角和右下角两个位置,计算中间最大值
复杂度O(n^4)
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=105;
const int inf=0x3f3f3f3f;
int a[N][N] , b[N][N] ;
int main(){
int n ;
scanf("%d" , &n) ;
for(int i = 1 ; i <= n ; i++){
for(int j = 1 ; j <= n ; j++){
scanf("%d" , &a[i][j]) ;
}
}
for(int i = 1 ; i <= n ; i++){
for(int j = 1 ; j <= n ; j++){
b[i][j] = b[i-1][j] + b[i][j-1] + a[i][j] - b[i-1][j-1] ;
}
}
int ans = -inf ;
for(int i = 0 ; i <= n ; i++){
for(int j = 0 ; j <= n ; j++){
for(int x = i+1 ; x <= n ; x++){
for(int y = j+1 ; y <= n ; y++){
ans = max(ans , b[x][y] - b[i][y] - b[x][j] + b[i][j]) ;
}
}
}
}
cout<<ans<<endl;
}
状态压缩dp做法
用暴力做法可以发现其实每次计算的时候每个子矩阵会有很多重复的位置
因此想办法把这些位置消掉
首先预处理每一列的前缀和,然后直接枚举子矩阵对应的上顶和下底对应的行数,在枚举列数,根据贪心思想计算就可以了
复杂度O(n^3)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110;
int a[N][N]={0};
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
cin>>a[i][j];
a[i][j]+=a[i-1][j];
}
int res=-0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
int last=0;
for(int k=1;k<=n;k++)
{
last=max(0,last)+a[j][k]-a[i-1][k];
res=max(last,res);
}
}
}
cout<<res<<endl;
}
4.最大子矩阵—两列数中选k个子矩阵和最大
题目链接
题解摘自:qcjj题解
m=1是最大连续k段和
m=2时:f[i][j][k]
表示第一列的前i
个数,第二列的前j
个数总共选了k
个子矩形的答案,枚举最后一个区间是怎么选的:
(再用一个sum[i][1/2]
表示第1/2
列的前缀和)
i==j
时,可能能有两列选到一个矩形的情况:
f[i][j][k] = max(f[x][x][k] + sum[i][1] - sum[x-1][1] + sum[i][2] - sum[x-1][2])
仅在第一列选一个区间:
f[i][j][k]=max(f[x][j][k-1]+sum[i][1]-sum[x-1][1])
仅在第二列选一个区间:
f[i][j][k]=max(f[i][x][k-1]+sum[j][2]-sum[x-1][2])
什么都不选:
f[i][j][k]=max(f[i-1][j][k],f[i][j-1][k])
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 110;
int n,m,k,t;
int f[N][N][15],a[N][3];
int main(){
cin>>n>>m>>t;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>k;
a[i][j]=a[i-1][j]+k;//每一列的前缀和
}
}
for(int i=1;i<=n;i++){//第一列
for(int j=1;j<=n;j++){//第二列
for(int k=1;k<=t;k++){
f[i][j][k]=max(f[i-1][j][k],f[i][j-1][k]);//都不选时
//只在第1列选时
for(int x=0;x<i;x++) f[i][j][k]=max(f[i][j][k],f[x][j][k-1]+a[i][1]-a[x][1]);
//只在第2列选时
for(int y=0;y<j;y++) f[i][j][k]=max(f[i][j][k],f[i][y][k-1]+a[j][2]-a[y][2]);
if(i!=j) continue;
//如果i==j,可能会出现在两列选一个矩形的情况
for(int x=0;x<i;x++) f[i][j][k]=max(f[i][j][k],f[x][x][k-1]+a[i][1]-a[x][1]+a[i][2]-a[x][2]);
}
}
}
cout<<f[n][n][t]<<endl;
}