hdu 6052单调栈 双指针

考虑每一种颜色的贡献, 考虑当前颜色时，该颜色的格子为1，其他格子为0，和不为0的子矩阵个数即为该颜色的贡献。
把每个颜色的点按上到下，左到右遍历，对每个点计算包含当前点，但不包含之前访问过得同颜色点的子矩阵个数。
访问(i,j)时候，下届可以任意取，枚举上界，从单调栈找出该上界的左右边界li, ri,弄两个指针搞之。 
该点答案贡献中加上(j - li ) * (ri - j) * (n - i + 1)

最后除以所有矩阵数 。

写的好cuo，调一年。。

#include 
#include 
#include 
#include 
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 };	//
		while (top && tmp.row >= stack[top - 1].row){
			top--;
		}
		stack[top++] = tmp;
	}
	return;
}
void Count2(int i, int j, int cc)
{
	topr = 0;
	stackr[topr++] = Node{ i, m + 1 };
	for (int k = m; k >= j; k--){
		Node tmp{ pre[cc][k], k };		//
		while (topr && tmp.row >= stackr[topr - 1].row){	//
			topr--;
		}
		stackr[topr++] = tmp;	//
	}
	return;
}
int main()
{
	int T;
	scanf("%d", &T);
	while (T--){
		init();
		scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++){
			for (int j = 1; j <= m; j++){
				scanf("%d", &color[i][j]);
			}
		}
		ll ans = 0;	//
		for (int i = 1; i <= n; i++){
			for (int j = 1; j <= m; j++){
				Count(i, j, color[i][j]);	//处理单调栈，左边行递减
				Count2(i, j, color[i][j]);	//右边行递增
				int l1 = 0, l2 = 0;
				for (int k = i; k > pre[color[i][j]][j]; k--){  //枚举上界
					while (!(stack[l1].row >= k && (l1 + 1 == top||stack[l1 + 1].row < k)))
						l1++;	//
					while (!(stackr[l2].row >= k && (l2 + 1 == topr || stackr[l2 + 1].row < k)))
						l2++;	//
					ans += (ll)(j - stack[l1].col) * (ll)(stackr[l2].col - j) * (ll)(n - i + 1);	
					// 左边可取列数 * 右边可取列数 * 下边可取行数
				}
				pre[color[i][j]][j] = i; //因为同一行右边不为访问过节点，所以后边更新pre
			}
		}
		printf("%.9lf\n", 4.0 * ans / (1.0 * n * (n + 1) * m * (m + 1)));
	}
	return 0;
}