最小子矩阵问题http://ac.jobdu.com/problem.php?pid=1139,是求最优解的一类题,看到题目就想到两种做法,一个是dp,一个是搜索遍历的方法。虽然深搜很多时候并不是用来解决最优解问题的,但是dp的做法,我暂时没有想到该怎么做,于是就先用DFS,代码如下:
#include <stdio.h>
using namespace std;
#define INF -999999999
int a[100][100],ans,n;
void DFS(int x0, int y0, int x, int y, int sum)
{
int temp;
if (x+1<n)
{
temp=0;
for (int i=y0;i<=y;++i) temp+=a[x+1][i];
temp+=sum;
if (temp>ans) ans=temp;
DFS(x0,y0,x+1,y,temp);
}
if (y+1<n)
{
temp=0;
for (int i=x0;i<=x;++i) temp+=a[i][y+1];
temp+=sum;
if (temp>ans) ans=temp;
DFS(x0,y0,x,y+1,temp);
}
return;
}
int main()
{
//freopen("input.txt","r",stdin);
//freopen("output.txt","w",stdout);
while (scanf("%d",&n)!=EOF)
{
for (int i=0;i<n;++i)
for (int j=0;j<n;++j)
scanf("%d",&a[i][j]);
ans=INF;
for (int i=0;i<n;++i)
for (int j=0;j<n;++j)
DFS(i,j,i,j,a[i][j]);
printf("%d\n",ans);
}
return 0;
}
这个做法可以解出样例,但是时间复杂度不符合要求,提交后超时。于是去看了fripside大神的做法,果然是用dp,而且基本思想就是将二维的子矩阵问题转化为一维的子序列问题,这种化高维为低维的方法,非常值得学习。目前我学过的dp类型,主要是最长递增子序列、最长公共子序列、背包问题等,如果见到想要用dp解决的问题,就要尽量往上面靠。为了独立完成代码,我用了与fripside的方法略不同的做法,他将矩阵按列求和化为序列,而我将矩阵按行求和,同时也证明的这两种化二维为一维的方法等价性。按行求和时,举出所有可能的求和方式,然后在求所得序列的最大子序列,时间复杂度为O(n^3)。但不知道为什么深搜会超时。AC代码如下:
#include <iostream>
using namespace std;
#define INF -999999999
int a[101][101];
int main()
{
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
int n;
while (cin>>n)
{
for (int i=1;i<=n;++i)
for (int j=0;j<=n;++j)
{
if (j==0) a[i][j]=0;
else
{
int x;
cin>>x;
a[i][j]=a[i][j-1]+x;
}
}
int ans=INF,temp;
for (int i=1;i<=n;++i)
for (int j=i;j<=n;++j)
{
int sum=0;
for (int k=1;k<=n;++k)
{
temp=a[k][j]-a[k][i-1];
if (sum>0) sum+=temp;
else sum=temp;
if (sum>ans) ans=sum;
}
}
cout<<ans<<endl;
}
return 0;
}