最大子矩阵(普通和01)
文章目錄
- 普通矩陣(單個矩陣值為任何數)
- 最大子段和
- 擴展到二維情況
- 01矩陣(單個矩陣值為0或1)
- 代碼:
普通矩陣(單個矩陣值為任何數)
例題:POJ 1074
求出其中最大的子矩陣
答案是:
9 2
-4 1
-1 8
最大和是15
我們先想想如果不是矩陣,是一個數組,求其中連續的最長一段,咋做?
最大子段和
我們用b[i]來表示a[0]…a[1]的最大子段和
那么b [ i ] =max ( b [i - 1 ] + a [ i ], a [ i ] )
當a[n] 大于dp[n-1](前n-1個元素序列的最大子段和)。此時舍棄前面的子段和,即斷開。
否則的話,把a[n]連接到前n-1個元素序列的最大和子段上,即dp[n] = dp[n-1] + a[n]
擴展到二維情況
二維矩陣排列方式如圖
參考題解
如果紅色是我們要找的最大子矩陣
那和就是:(a[q][i]+…+a[p][i],a[q][i+1]+…+a[p][i+1],…,a[q][j]+…a[p][j])=(sum1,sum2,…,sumn)
sum也就是每一列的相同行數相加
這不也是sun的一維數組,那其實就是一個一維數組的最大子段問題
如果把二維數組看成是縱向的一維數組和橫向的一維數組
01矩陣(單個矩陣值為0或1)
參考題解
POJ - 3494
其實就是沒有負數情況
我一開始想的是將最大子矩陣的代碼中,如果一個數為0就賦值為-10000(這樣程序就肯定不會選擇這個塊)
但是這樣會超時
針對01矩陣我們可以這么想:
每一個單元格的值等于它所在列的連續1的數量,如果當前行為1,則當前行的值就等于上一行加1,如果當前行為0,則他的值也是0
我們掃描每一行,求出這一行中可形成的矩形最大面積,然后去最大情況即可
然后我們要用到一個單調遞減的單調棧,
如果棧為空或者當前元素大于等于棧頂元素,則入棧
若棧非空且當前元素小于棧頂元素,則將棧頂彈出,更新面積最大值,直到棧空或者遇到第一個大于等于當前元素
將最后一個出棧的棧頂元素(最后一個大于當前元素的)向左右延伸,修改其值,并入棧。
核心都是對于每個高度為h的矩形,找到最左高度大于等于它的位置,和最右的位置。
我們看第三行:
0 1 3 2 0 0 0
0先入棧,然后1入棧,然后3入棧,然后到2時,2<3,計算此時的矩陣面積=(4-3) * 3=3,此時最大面積是3,然后將值3改為2,此時就是0 1 2 2 ,然后0<2,計算此時面積 = (5-3) * 2=4,最大面積就是4,一次類推,都是按照這個操作
代碼:
#include<stdio.h> #include<string.h> #include<iostream> #include<stack> using namespace std;int main() {//top指向棧頂;tmp為臨時變量,記錄面積的值;ans為答案,記錄面積的最大值 int i,j,m,n,x,top,tmp,ans,h[2020],a[2020];stack<int> st; //單調棧,記錄位置 while(~scanf("%d%d",&m,&n)){ans=0;memset(h,0,sizeof(h)); //用于第一行的處理for(i=0;i<m;i++){ //掃描每一行 for(j=1;j<=n;j++){scanf("%d",&x);if(x==1) h[j]=h[j]+1; //如果是1,則在上一行本列的基礎上+1 else h[j]=0; //否則為0 a[j]=h[j]; //a數組用來向左右擴展}a[n+1]=-1; //設最后元素為最小值,以最后讓棧內元素出棧 for(j=1;j<=n+1;j++){if(st.empty()||a[j]>=a[st.top()]){ //如果棧為空或入棧元素大于等于棧頂元素,則入棧 st.push(j);}else{while(!st.empty()&&a[j]<a[st.top()]){ //如果棧非空并且入棧元素小于棧頂元素,則將棧頂元素出棧 top=st.top();st.pop();tmp=(j-top)*a[top]; //計算面積值 if(tmp>ans) ans=tmp; //更新面積最大值 }st.push(top); //將最后一次出棧的棧頂元素延伸并入棧 a[top]=a[j]; //修改其對應的值 }}}printf("%d\n",ans);}return 0; }總結
以上是生活随笔為你收集整理的最大子矩阵(普通和01)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 00003 不思议迷宫.0006:客户端
- 下一篇: Poj 1011 UVA - 307 S