hdu 6052单调栈 双指针

来源:互联网 发布:卡在windows启动界面 编辑:程序博客网 时间:2024/05/01 06:47

考虑每一种颜色的贡献, 考虑当前颜色时,该颜色的格子为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;}


原创粉丝点击