hdu 6052單調棧 雙指標
阿新 • • 發佈:2019-01-06
考慮每一種顏色的貢獻, 考慮當前顏色時,該顏色的格子為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; }