（来源：XJ高质量原创题）
弱化版
题意：\(n\times n\) 的棋盘上有 \(n\) 颗棋子，每行每列都有且仅有一颗棋子，求出有多少个 \(k\times k\) 的子棋盘也满足每行每列只有一颗棋子。
将棋盘的 \(x\) 轴看作上一题中每个点的下标，\(y\) 轴看作这个点的权值。
其实就是树退化为了链，而 \(m=1\) 的情况。
题意
给定一棵树，每一个点有一个权值 \(a_i\) ，\(\{a\}\) 构成一个排列。
询问有多少个点集 \(S={u_1,u_2,\dots,u_k}(1\le j\le n)\) 满足这些点构成联通块不超过 \(m\) 个，且 \(\max_{1\le i\le k}\{a_{u_{i}}\}-\min_{1\le i\le k}\{a_{u_{i}}\}+1=k\) 。
\(n\le 10^5,m\le 7\)
题解
考虑枚举值域的右端点 \(r\)，考虑加入这个右端点时，对与 \([1,r-1]\) 中的最短点的联通块个数的变化。
先假设假设这个点与其他无边相连，那么联通块数加 \(1\) 。
对于每一条与点集中的点相连的边 \((num(r),ver)\) ，都会使左端点在 \([1,val(ver)]\) 中的联通块数量 \(-1\) （\(num(x)\) 表示权值为 \(x\) 的点的编号）。
之后我们就将题意简化为：
区间加上一个数
求出整个数列中 \(\le m\) 的数有多少个
保证 \(a[1]=1\)
考虑到 \(m\) 比较小，猜测答案复杂度为 \(O(nm\log n)\) 。
我们将加减操作同普通的线段树操作，主要改变 \(\text{pushup}\) 操作。
在线段树的每一个节点 \([l,r]\) 上维护：
\(minn\) ： 表示在 \([l,r]\) 区间上数的最小值。
\(cnt[7]\) ： 表示这段区间内值域在 \([minn,minn+6]\) 之间的数分别有多少。
之后根据两个子区间的最小值就可以愉快地转移啦！！
// Author:A weak man named EricQian
// expect : 100 pts
#include<bits/stdc++.h>
using namespace std;
#define infll 0x7f7f7f7f7f7f7f7f
#define inf 0x3f3f3f3f
#define Maxn 500005
#define Maxm 7
typedef long long ll;
inline int rd()
{
int x=0;
char ch,t=0;
while(!isdigit(ch = getchar())) t|=ch=='-';
while(isdigit(ch)) x=x*10+(ch^48),ch=getchar();
return x=t?-x:x;
}
int n,m,tot;
int val[Maxn],num[Maxn];
int hea[Maxn],nex[Maxn<<1],ver[Maxn<<1];
struct TREE { ll minn,laz,cnt[Maxm]; }tree[Maxn<<2];
ll ans;
inline void add(int x,int y){ ver[++tot]=y,nex[tot]=hea[x],hea[x]=tot; }
inline void pushup(int p)
{
int x=p<<1,y=p<<1|1,tmp;
if(tree[x].minn>tree[y].minn) swap(x,y);
tree[p].minn=tree[x].minn;
for(int i=0;i<m;i++) tree[p].cnt[i]=tree[x].cnt[i];
tmp=tree[y].minn-tree[x].minn;
for(int i=tmp;i<m;i++) tree[p].cnt[i]+=tree[y].cnt[i-tmp];
}
inline void pushdown(int p)
{
tree[p<<1].laz+=tree[p].laz,tree[p<<1].minn+=tree[p].laz;
tree[p<<1|1].laz+=tree[p].laz,tree[p<<1|1].minn+=tree[p].laz;
tree[p].laz=0;
}
void build(int p,int nl,int nr)
{
if(nl==nr) { tree[p].minn=tree[p].cnt[0]=1; return; }
int mid=(nl+nr)>>1;
build(p<<1,nl,mid),build(p<<1|1,mid+1,nr);
pushup(p);
}
void add(int p,int nl,int nr,int l,int r,int k)
{
if(nl>=l && nr<=r) { tree[p].minn+=k,tree[p].laz+=k; return; }
pushdown(p);
int mid=(nl+nr)>>1;
if(mid>=l) add(p<<1,nl,mid,l,r,k);
if(mid<r) add(p<<1|1,mid+1,nr,l,r,k);
pushup(p);
}
int main()
{
n=rd(),m=rd();
for(int i=1;i<=n;i++) val[i]=rd(),num[val[i]]=i;
for(int i=1,x,y;i<n;i++) x=rd(),y=rd(),add(x,y),add(y,x);
build(1,1,n);
for(int r=1;r<=n;r++)
{
if(r!=1) add(1,1,n,1,r-1,1);
for(int i=hea[num[r]];i;i=nex[i])
if(val[ver[i]]<val[num[r]])
add(1,1,n,1,val[ver[i]],-1);
for(int i=0;i<m;i++) ans+=tree[1].cnt[i];
ans-=(n-r);
}
printf("%lld\n",ans);
return 0;
}