构成一棵树可以分成两个限制：图不成环、图的点数-边数=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;
}