力扣1878. 矩阵中最大的三个菱形和



这一题的大意是指给出一个矩阵,里面的点可以构成菱形,这个菱形的特点是由正方形旋转45度构成的,菱形的面积也可以为0,那么它就可能是由一个单个的点构成。

现在让我们找在这个矩阵中,边长所经过的点的和最大的三个菱形。

我的思路是我们只需要枚举出来符合条件的菱形,然后计算各边之和放入数组中排序,即可得到点之和最大的三个菱形。

这实际上是类似在矩阵中枚举子矩阵的问题,无非是现在是来枚举菱形

类似题目:

力扣1895. 最大的幻方
力扣2132. 用邮票贴满网格图

因为这一题的数据范围比较小,只有10^2,所以可以大胆的尝试一下暴力:

枚举边长大约需要O(n),枚举矩阵中的所有点需要O(n^m)约等于O(n ^2),

在计算菱形上的所有点的和大概最长才为O(min(n,m)*4) 更何况大多数的边长情况,远远不能得到,因此是小于10^8的。

暴力写法事实也可以通过:

cpp 复制代码
class Solution {
public:
static bool cmp(int a,int b)
{
	return a>b;
}
  vector<int> getBiggestThree(vector<vector<int>>& grid) {
        int m=grid.size();
        int n=grid[0].size();
        vector<int> ans;
        vector<int> temp;
       for(int edge=0;edge<min(n,m);edge++)
       {
       	   for(int i=0;i<m;i++)
       	   {
       	   	for(int j=0;j<n;j++)
       	   	{
       	   		int x1=i;
       	   		int y1=j;
       	   		int x2=i+edge;
       	   		int y2=j-edge;
       	   		int x3=i+edge*2;
       	   		int y3=j;
       	   		int x4=i+edge;
       	   		int y4=j+edge;
       	   		if(x2>=m||y2<0||x3>=m||x4>=m||y4>=n)
       	   		{
       	   			continue;
				}
				int sum=0;
			       for(int x=x2,y=y2;x>=x1;x--,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x2+1,y=y2+1;x<=x3;x++,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x1+1,y=y1+1;x<=x4;x++,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x3-1,y=y3+1;x>=x4+1;x--,y++)
					{
					     sum+=grid[x][y];	
					} 
					//cout<<sum<<endl;
					ans.push_back(sum);
			}
		   }
	   }
        sort(ans.begin(),ans.end(),cmp);
        int t; 
      
        for(int i=0;i<ans.size();i++)
        {
        	if(i==0)
        	{
        	temp.push_back(ans[i]);
			t=ans[i];
				
			}
			else if(t==ans[i])
			{
				continue;
			}
			else
			{
				temp.push_back(ans[i]);
				t=ans[i];
			}
			if(temp.size()==3)
			{
				break;
			}
        	
		}
        return temp;
        
        
        
        
    }
};

也可以先枚举点,再枚举边,思路实际是一样的,代码写法有点差异

cpp 复制代码
class Solution {
public:
static bool cmp(int a,int b)
{
	return a>b;
}
  vector<int> getBiggestThree(vector<vector<int>>& grid) {
        int m=grid.size();
        int n=grid[0].size();
        vector<int> ans;
        vector<int> temp;
        for(int i=0;i<m;i++)
        {
        	for(int j=0;j<n;j++)
        	{
        		//这是枚举了一个点
				//现在我们要看从这个点可以引出怎么样的矩形
			     ans.push_back(grid[i][j]);
				for(int k=1;k<min(m,n);k++)
				{
					//已知一个顶点和边长,自然可以推出这个菱形
					int x1=i;
					int y1=j;
					int x2=i+k;
					int y2=j-k;
					int x3=i+2*k;
					int y3=j;
					int x4=i+k;
					int y4=j+k; 
					if(y2<0||x3>=m||x4>=m||y4>=n)
					{
						break;
					}
					//当枚举出来矩阵后我们需要计算这个矩阵的值
					int sum=0;
					for(int x=x2,y=y2;x>=x1;x--,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x2+1,y=y2+1;x<=x3;x++,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x1+1,y=y1+1;x<=x4;x++,y++)
					{
						sum+=grid[x][y];
					}
					for(int x=x3-1,y=y3+1;x>=x4+1;x--,y++)
					{
					     sum+=grid[x][y];	
					} 
					ans.push_back(sum);
				 } 
			}
		}
       
        sort(ans.begin(),ans.end(),cmp);
        int t; 
      
        for(int i=0;i<ans.size();i++)
        {
        	if(i==0)
        	{
        	temp.push_back(ans[i]);
				t=ans[i];
				
			}
			else if(t==ans[i])
			{
				continue;
			}
			else
			{
				temp.push_back(ans[i]);
				t=ans[i];
			}
			if(temp.size()==3)
			{
				break;
			}
        	
		}
        return temp;
        
        
        
        
    }
};

那么和在矩阵中枚举子矩阵一样,在计算边长的时候,我们可以用前缀和进行优化,提前算出边长,这样时间复杂度大概为O(n^3)一定可以过的。

但一般的前缀和是好求的,可这一题是菱形,菱形的边长是斜着的,因此用斜向的前缀和来求。

斜向前缀和怎么求的呢?

实际上就是求主对角线和副对角线的前缀和

也即求:

从左上到右下 ↘ 的前缀和

从右上到左下 ↙ 的前缀和

cpp 复制代码
        for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=n;j++)
			{
				sum1[i][j]=sum1[i-1][j-1]+grid[i-1][j-1];
				sum2[i][j]=sum2[i-1][j+1]+grid[i-1][j-1];
			}
		 } 

注意sum2数组的列要开大一点,因为在求从右上到左下的前缀和时,j+1可能会越界。

我们画画图就可以弄清楚:

当我们知道怎么求前缀和之和,那么就可以把原来需要枚举的过程,用前缀和来表示,从而降低时间复杂度。

cpp 复制代码
class Solution {
public:
static bool cmp(int a,int b)
{
	return a>b;
}
  vector<int> getBiggestThree(vector<vector<int>>& grid) {
        int m=grid.size();
        int n=grid[0].size();
        vector<int> ans;
        vector<int> temp;
        vector<vector<int> > sum1(m+1,vector<int>(n+1+1,0));
		vector<vector<int> > sum2(m+1,vector<int>(n+1+1,0));
		for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=n;j++)
			{
				sum1[i][j]=sum1[i-1][j-1]+grid[i-1][j-1];
				sum2[i][j]=sum2[i-1][j+1]+grid[i-1][j-1];
			}
		 } 
       for(int edge=0;edge<max(n,m);edge++)
       {
       	   for(int i=0;i<m;i++)
       	   {
       	   	for(int j=0;j<n;j++)
       	   	{
       	   		int x1=i;
       	   		int y1=j;
       	   		int x2=i+edge;
       	   		int y2=j-edge;
       	   		int x3=i+edge*2;
       	   		int y3=j;
       	   		int x4=i+edge;
       	   		int y4=j+edge;
       	   		if(x2>=m||y2<0||x3>=m||x4>=m||y4>=n)
       	   		{
       	   			continue;
				}
				int sum=0;
				if(edge==0)
				{
					ans.push_back(grid[i][j]);
					continue;
				}
				else
				{
					sum+=sum2[x2+1][y2+1]-sum2[x1][y1+2];
			    sum+=sum2[x3+1][y3+1]-sum2[x4][y4+2];
			    sum+=sum1[x4+1][y4+1]-sum1[x1][y1];
			    sum+=sum1[x3+1][y3+1]-sum1[x2][y2];
			    sum-=(grid[x1][y1]+grid[x2][y2]+grid[x3][y3]+grid[x4][y4]);
				ans.push_back(sum);
				}
				
			}
		   }
	   }
        sort(ans.begin(),ans.end(),cmp);
        int t; 
      
        for(int i=0;i<ans.size();i++)
        {
        	if(i==0)
        	{
        	temp.push_back(ans[i]);
			t=ans[i];
				
			}
			else if(t==ans[i])
			{
				continue;
			}
			else
			{
				temp.push_back(ans[i]);
				t=ans[i];
			}
			if(temp.size()==3)
			{
				break;
			}
        	
		}
        return temp;
        
        
        
        
    }
};

这里和暴力枚举的过程有不同的是,我们无法表示edge=0的情况,因为在该情况下,用前缀和计算时会有越界。因此我们需要把edge==0的情况单独讨论。

总结:

这一题还是典型的在矩阵中枚举"子矩阵"的一类问题,无非这里的子矩阵是菱形,在优化时需要用斜向前缀和,但思路都是类似的。

优化后的时间复杂度

O(n^3)

相关推荐
Miraitowa_cheems2 小时前
LeetCode算法日记 - Day 73: 最小路径和、地下城游戏
数据结构·算法·leetcode·职场和发展·深度优先·动态规划·推荐算法
野蛮人6号2 小时前
力扣热题100道之560和位K的子数组
数据结构·算法·leetcode
Swift社区3 小时前
LeetCode 400 - 第 N 位数字
算法·leetcode·职场和发展
fengfuyao9853 小时前
BCH码编译码仿真与误码率性能分析
算法
小白不想白a4 小时前
每日手撕算法--哈希映射/链表存储数求和
数据结构·算法
剪一朵云爱着4 小时前
力扣2080. 区间内查询数字的频率
算法·leetcode
落日漫游4 小时前
数据结构笔试核心考点
java·开发语言·算法
Dream it possible!4 小时前
LeetCode 面试经典 150_栈_有效的括号(52_20_C++_简单)(栈+哈希表)
c++·leetcode·面试··哈希表
workflower5 小时前
Fundamentals of Architectural Styles and patterns
开发语言·算法·django·bug·结对编程
仰泳的熊猫5 小时前
LeetCode:701. 二叉搜索树中的插入操作
数据结构·c++·算法·leetcode