传送门

构成一棵树可以分成两个限制：图不成环、图的点数-边数=1。

我们考虑枚举右端点\(r\)计算所有可能的左端点\(l\)的答案。我们先考虑第一个限制：图不成环。注意到当\(r\)确定的时候，满足这个条件的\(l\)一定是一段后缀。设\(p_r\)表示满足图不成环时最小的\(l\)，还可以发现\(p_r\)是单调不降的。那么我们可以使用双指针维护，在\(r\)增加\(1\)的时候使用LCT维护\(p_r\)的值。

接下来在每一个\(p_r\)求完之后，考虑图的点数-边数=1的限制。每一次\(r\)增加\(1\)的时候，都会增加一些新的可能的边。找到在矩阵上与\(r\)相邻且值在\([l,r]\)内的元素，设其中某一个的元素值为\(x\)，那么在\(l \leq x\)的时候边\((x,r)\)就会出现。我们可以使用线段树维护每一个左端点的点数-边数，每一次增加一些新点和新边就是前缀加减的过程。

最后我们需要查询线段树中值为\(1\)的数的个数。这在区间加法下似乎不太可做，但是注意到如果将不合法位置的值设为INF，那么线段树中\(1\)一定是最小值。所以就相当于是一个求区间最小值数量的问题，就可以使用线段树去做了。

#include<bits/stdc++.h>
using namespace std;int read(){int a = 0; char c = getchar(); bool f = 0;while(!isdigit(c)){f = c == '-'; c = getchar();}while(isdigit(c)){a = a * 10 + c - 48; c = getchar();}return f ? -a : a;
}#define id(i , j) ((i - 1) * M + j)
#define PII pair < int , int >
const int _ = 2e5 + 3 , dir[4][2] = {0,1,0,-1,1,0,-1,0};
int N , M , L = 1 , R , arr[_]; long long cnt; PII to[_];namespace segt{int mn[_ << 2] , cnt[_ << 2] , mrk[_ << 2];#define mid ((l + r) >> 1)
#define lch (x << 1)
#define rch (x << 1 | 1)void init(int x , int l , int r){mn[x] = 1e9; cnt[x] = r - l + 1; if(l != r){init(lch , l , mid); init(rch , mid + 1 , r);}}void mark(int x , int val){mrk[x] += val; mn[x] += val;}void down(int x){mark(lch , mrk[x]); mark(rch , mrk[x]); mrk[x] = 0;}void up(int x){mn[x] = min(mn[lch] , mn[rch]); cnt[x] = (mn[x] == mn[lch]) * cnt[lch] + (mn[x] == mn[rch]) * cnt[rch];}void modify(int x , int l , int r , int L , int R , int val){if(l >= L && r <= R) return mark(x , val);down(x);if(mid >= L) modify(lch , l , mid , L , R , val);if(mid < R) modify(rch , mid + 1 , r , L , R , val);up(x);}
}namespace LCT{int fa[_] , ch[_][2]; bool rmrk[_];bool nroot(int x){return ch[fa[x]][0] == x || ch[fa[x]][1] == x;}bool son(int x){return ch[fa[x]][1] == x;}void mark(int x){rmrk[x] ^= 1; swap(ch[x][0] , ch[x][1]);}void down(int x){if(rmrk[x]){mark(ch[x][0]); mark(ch[x][1]); rmrk[x] = 0;}}void dall(int x){if(nroot(x)) dall(fa[x]); down(x);}void rot(int x){bool f = son(x); int y = fa[x] , z = fa[y] , w = ch[x][f ^ 1];fa[x] = z; if(nroot(y)) ch[z][son(y)] = x;fa[y] = x; ch[x][f ^ 1] = y;ch[y][f] = w; if(w) fa[w] = y;}void splay(int x){dall(x); while(nroot(x)){if(nroot(fa[x])) rot(son(fa[x]) == son(x) ? fa[x] : x); rot(x);}}void access(int x){for(int y = 0 ; x ; y = x , x = fa[x]){splay(x); ch[x][1] = y;}}int fdrt(int x){access(x); splay(x); while(ch[x][0]) down(x = ch[x][0]); splay(x); return x;}void mkrt(int x){access(x); splay(x); mark(x);}void split(int x , int y){mkrt(x); access(y); splay(y);}void link(int x , int y){mkrt(x); fa[x] = y;}void cut(int x , int y){split(x , y); ch[y][0] = fa[x] = 0;}
}void cut(){PII pos = to[L]; segt::modify(1 , 1 , N * M , L , L , 1e9);for(int i = 0 ; i < 4 ; ++i){int x = pos.first + dir[i][0] , y = pos.second + dir[i][1];if(x > 0 && x <= N && y > 0 && y <= M && LCT::fdrt(arr[id(x , y)]) == LCT::fdrt(L))LCT::cut(arr[id(x , y)] , L);}++L;
}void link(){PII pos = to[R];for(int i = 0 ; i < 4 ; ++i){int x = pos.first + dir[i][0] , y = pos.second + dir[i][1];if(x > 0 && x <= N && y > 0 && y <= M){int t = arr[id(x , y)];while(t >= L && LCT::fdrt(t) == LCT::fdrt(R)) cut();if(t >= L && t <= R) LCT::link(t , R);}}for(int i = 0 ; i < 4 ; ++i){int x = pos.first + dir[i][0] , y = pos.second + dir[i][1];if(x > 0 && x <= N && y > 0 && y <= M && arr[id(x , y)] >= L && arr[id(x , y)] <= R)segt::modify(1 , 1 , N * M , 1 , arr[id(x , y)] , -1);}
}int main(){N = read(); M = read(); segt::init(1 , 1 , N * M);for(int i = 1 ; i <= N ; ++i) for(int j = 1 ; j <= M ; ++j) to[arr[id(i , j)] = read()] = PII(i , j);while(++R <= N * M){segt::modify(1 , 1 , N * M , R , R , -1e9);segt::modify(1 , 1 , N * M , 1 , R , 1); link();cnt += (segt::mn[1] == 1) * segt::cnt[1];}cout << cnt; return 0;
}