LeetCode 554. 砖墙（C++、python）

你的面前有一堵方形的、由多行砖块组成的砖墙。 这些砖块高度相同但是宽度不同。你现在要画一条自顶向下的、穿过最少砖块的垂线。

砖墙由行的列表表示。 每一行都是一个代表从左至右每块砖的宽度的整数列表。

如果你画的线只是从砖块的边缘经过，就不算穿过这块砖。你需要找出怎样画才能使这条线穿过的砖块数量最少，并且返回穿过的砖块数量。

你不能沿着墙的两个垂直边缘之一画线，这样显然是没有穿过一块砖的。

 

示例：

输入: [[1,2,2,1],
      [3,1,2],
      [1,3,2],
      [2,4],
      [3,1,2],
      [1,3,1,1]]

输出: 2

解释: 

 

提示：

每一行砖块的宽度之和应该相等，并且不能超过 INT_MAX。

每一行砖块的数量在 [1,10,000] 范围内， 墙的高度在 [1,10,000] 范围内， 总的砖块数量不超过 20,000。

C++

class Solution {
public:
    int leastBricks(vector<vector<int>>& wall) 
    {
        int m=wall.size();
        unordered_map<int,int> tmp;
        for(int i=0;i<m;i++)
        {
            int sum=0;
            int n=wall[i].size();
            for(int j=0;j<n-1;j++)
            {
                sum+=wall[i][j];
                tmp[sum]++;
            }
        }
        int cc=0;
        unordered_map<int,int>::iterator it;
        for(it=tmp.begin();it!=tmp.end();it++)
        {
            cc=max(cc,it->second);
        }  
        return m-cc;  
    }
};

python

class Solution:
    def leastBricks(self, wall):
        """
        :type wall: List[List[int]]
        :rtype: int
        """
        m=len(wall)
        dic={}
        for i in range(m):
            su=0
            for j in range(0,len(wall[i])-1):
                su+=wall[i][j]
                if su not in dic:
                    dic[su]=1
                else:
                    dic[su]+=1
        cc=0
        for key in dic:
            if dic[key]>cc:
                cc=dic[key]
        return m-cc