您的位置:首页 > 编程语言 > C语言/C++

Leetcode Maximal Square

2015-09-16 02:32 435 查看
Leetcode Maximal Square 相关代码,本算法使用dp算法完成,主要思想为:一列列的计算,计算每个元素右上解的最大square,并记录以其为右下角的square的信息(左边’1’的长度、以及相应square大小)。计算时,使用其左边的元素的信息进行相关计算,具体如下:

如果从上到元素相邻接的 ‘1’大于左边元素为右下角的square的边长,则判断当前元素为右下角的square边长能否增长1,如果能,应该有当前square的上边,在左侧元素的上边之上,如下:

xxxxx

yyyyx

yyyyx

yyyyx

其中x代表为新增的元素,其应为1, y为左侧元素为右下角的square。

注:此处纵向x为计算ajacent,横向x为记录左侧’1’的长度。

算法复杂度为o(n^2)。

相关代码与测试如下:

#include <iostream>
#include <vector>
using namespace std;

// conclusion: use the left rectangle and one colmun to calculate the current value
class Solution {
public:
int maximalSquare(vector<vector<char> >& matrix) {
if (matrix.size() == 0) {
return 0;
}
if (matrix[0].size() == 0) {
return 0;
}
int row_num = matrix.size();
int col_num = matrix[0].size();
vector<vector<int> > row(row_num, vector<int>(2, 0));
int max = 0;
int ajacent = 0;
for (int index = 0; index < col_num; index ++) {
ajacent = 0;
for (int j = 0; j < row_num; j ++) {
if (matrix[j][index] == '1') {
row[j][0] ++;
ajacent ++;
if (ajacent > row[j][1] && row[j - row[j][1]][0] >= row[j][1] + 1) {
row[j][1] ++;
max = row[j][1] > max? row[j][1] : max;
} else {
row[j][1] = ajacent > row[j][1]? row[j][1] : ajacent;
}
} else {
row[j][0] = 0;
ajacent = 0;
row[j][1] = 0;
}
}
}
return max;
}
};

int main(int argc, char * argv[]) {
Solution so;
vector<vector<char> > test(3, vector<char>(5, '1'));
test[0][1] = '0';
test[1][3] = '0';
test[2][0] = '0';
int re = so.maximalSquare(test);
cout<<"max:"<<re<<endl;
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode 算法 c++ dp