3972 方格集数量（组合计数）

1. 问题描述：

给定一个 n × m 的方格矩阵。每个方格要么是黑色，要么是白色。请你计算，共有多少个非空方格集合满足：
集合内的所有方格颜色都相同。
集合内的任意两个方格都在同一行或同一列上。

输入格式

第一行包含两个整数 n，m。接下来 n 行，每行包含 m 个整数，每个整数要么是 0，要么是 1，用来表示矩阵中每个方格的颜色，0 表示颜色为白，1 表示颜色为黑。

输出格式

一个整数，表示满足条件的非空方格集合数量。

数据范围

前四个测试点满足 1 ≤ n，m ≤ 10，
所有测试点满足 1 ≤ n，m ≤ 50。

输入样例1：

1 1
0

输出样例1：

1

输入样例2：

2 3
1 0 1
0 1 0

输出样例2：

8
来源：https://www.acwing.com/problem/content/3975/

2. 思路分析：

分析题目可以知道我们只能够在每一行或者每一列中选择，我们可以先考虑行，对于当前这一行若有a个黑色方格，则有m - a个白色方格，对于黑色方格可以选择的方案数目为2 ^ a - 1，白色格子可以选择的方案数目为2 ^ (m - a) - 1，减1是因为不能够选择空格子，对于每一行都可以这样统计，而且行与列是类似的，所以也可以使用同样的方法统计方案数目，但是这样统计会出现一些重复，对于每一个格子会在行与列的时候都统计到了，所以需要减去重复计算的n * m个格子。

3. 代码如下：

class Solution:
    def process(self):
        n, m = map(int, input().split())
        # row, col分别存储的是每一行, 每一列黑色格子的数量, 白色格子的数量可以使用对应行列的数目减去得到
        row, col = [0] * (n + 10), [0] * (m + 10)
        for i in range(n):
            a = list(map(int, input().split()))
            for j in range(m):
                if a[j] == 1:
                    row[i] += 1
                    col[j] += 1
        # 枚举所有的行
        res = 0
        for i in range(n):
            # 1 << k表示2 ^ k
            res += (1 << row[i]) - 1
            res += (1 << (m - row[i])) - 1
        # 枚举所有的列
        for i in range(m):
            res += (1 << col[i]) - 1
            res += (1 << n - col[i]) - 1
        # 因为每一个格子被多计算了一次, 所以需要减掉
        return res - n * m


if __name__ == '__main__':
    print(Solution().process())