考虑每一种颜色的贡献, 考虑当前颜色时,该颜色的格子为1,其他格子为0,和不为0的子矩阵个数即为该颜色的贡献。
把每个颜色的点按上到下,左到右遍历,对每个点计算包含当前点,但不包含之前访问过得同颜色点的子矩阵个数。
访问(i,j)时候,下届可以任意取,枚举上界,从单调栈找出该上界的左右边界li, ri,弄两个指针搞之。
该点答案贡献中加上(j - li ) * (ri - j) * (n - i + 1)
最后除以所有矩阵数 。
写的好cuo,调一年。。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
const int MAXN = 110;
int n, m, color[MAXN][MAXN];
struct Node
{
int row, col;
Node(){}
Node(int _i, int _j) :row(_i), col(_j){}
}stack[MAXN], stackr[MAXN];
int top, topr;
int pre[MAXN * MAXN][MAXN];
int now[MAXN * MAXN][MAXN];
void init()
{
memset(pre, 0, sizeof(pre));
}
void Count(int i, int j, int cc)
{
top = 0;
stack[top++] = Node{ i, 0 };
for (int k = 1; k <= j; k++){
Node tmp{ pre[cc][k], k };