HDU-6052 To my boyfriend（单调栈）

传送门：HDU-6052

对于每种颜色我们考虑它对答案的贡献，因为一种颜色可能会有多个方格，在计算某个方格的贡献时，与它相同颜色且被计算过的方格是不能考虑的，这时可以把这些方格当做障碍物。

具体做法是：设当前是第i行，pre[col][j]表示计算到第i-1行时，在第j列上，颜色为col的方格出现的位置，new[col][j]表示计算到第i行时，在第j列上，颜色为col的方格出现的位置。如果要计算[i,j]对答案的贡献，则扫描列为[1,m]，[1,j-1]，[j+1,m]的矩阵中矩形的个数cnt1,cnt2,cnt3（行为[1,i]）,贡献即为：(cnt1-cnt2-cnt3)*(n-i+1)

扫描用单调栈做，总复杂度为O(n^3)

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int MX = 105;
int n, m, mp[MX][MX];
int st[105], c[105], h[MX], L[MX];
int pre[MX*MX][MX];  //在第j列上一次出现的位置
int now[MX*MX][MX];  //在第j列这一次出现的位置
int col, x, y;
//单调栈计算矩阵数目
int solve(int l, int r) {
    memset(c, 0, sizeof(c));
    int ans = 0, sz = 0, sum = 0;
    st[sz] = l - 1;
    for (int k = l; k <= r; k++) {
        h[k] = (k < y || x != now[col][k]) ? x - now[col][k] : x - pre[col][k];
        while (sz && h[st[sz]] >= h[k]) {
            sum -= h[st[sz]] * (st[sz] - L[st[sz]]);
            sz--;
        }
        L[k] = st[sz];
        st[++sz] = k;
        sum += h[k] * (k - L[k]);
        ans += sum;
    }
    return ans;
}

int main() {
    //freopen("in.txt", "r", stdin);
    int T;
    scanf("%d", &T);
    while (T--) {
        memset(pre, 0, sizeof(pre));
        memset(now, 0, sizeof(now));
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= m; j++)
                scanf("%d", &mp[i][j]);
        LL ans = 0, mx = n * m * (n + 1) * (m + 1) / 4;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                pre[mp[i][j]][j] = now[mp[i][j]][j];
                now[mp[i][j]][j] = i;
            }
            for (int j = 1; j <= m; j++) {
                col = mp[i][j]; x = i; y = j;
                ans += (n - i + 1) * (solve(1, m) - solve(1, j - 1) - solve(j + 1, m));
            }
        }
        printf("%.9f\n", (double)ans / mx);
    }
}