题意:给定一个n行m列的矩阵,每个格子上有一个颜色,询问任意选择一个子矩阵,其颜色种类数的期望。n与m分范围为100。
解法:
我们定义矩阵的价值为该矩阵的颜色种类数。显然求期望的话,可以考虑求出所有子矩阵的价值之和,除以不同子矩阵的个数。
不同子矩阵的个数拿隔板法推一推可知是 (n+1)*n*(m+1)*m/4 的。
接着考虑求出所有子矩阵价值之和。
可以分别对每一种颜色进行处理。对于任一颜色,若想求出该颜色对所有子矩阵价值的贡献,可以将图看成是仅有黑白格点的图,黑点代表有颜色,白点代表无颜色,则有:贡献=总矩阵个数-全部都为白点的矩阵的个数。
求黑白格子全部都为白点的矩阵个数,涉及到一个经典题 hihocoder 1476。该题比较主流的做法有两种,一种是DP,一种是容斥:
1. DP:对于左上角点为(1, 1),右下角点为(n, m)的矩阵,定义该矩阵内仅有白点的子矩阵其种类数为dp[i][j],则dp[i][j]=dp[i-][j]+dp[i][j-1]-dp[i-1][j-1]+sum[以点(i, j)为右下角的全白矩阵个数],以点(i, j)为右下角的全白矩阵个数可以用单调栈维护该矩阵的左上角可以放的位置个数。这个算法在这题中最坏的复杂度是O(nnmm)的,即当有一半的格子是黑色的时候,不过对拍了8组矩阵里只有两种颜色的数据之后,速度很快,说明常数很小。
2.容斥:全部为白点的矩阵个数&