1. 程式人生 > >hdu 6052單調棧 雙指標

hdu 6052單調棧 雙指標

考慮每一種顏色的貢獻, 考慮當前顏色時,該顏色的格子為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 };	//
		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;
}