bzoj5392 [Lydsy1806月赛]路径统计

传送门

分析

我们设sum[x]为小于等于x的点现在有多少联通

于是一个序列合法当且只当sum[R]-sum[L-1]=len且所有点度数不大于2

我们知道如果对于序列[L,R]满足条件则[L+1,R]一定满足

如果[L,R]不满足则[L-1,R]一定不满足

所以我们可以枚举R然后找最靠左的满足度数都小于2的L

用线段树维护信息查询区间内最大值是R的数的个数就是贡献

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
int n,L,R,num,Max,cnt,col[],d[],sum[],du[];
long long Ans;
vector<int>high[],low[];
inline void build(int le,int ri,int wh){
    sum[wh]=;
    d[wh]=ri;
    if(le==ri)return;
    int mid=(le+ri)>>;
    build(le,mid,wh<<);
    build(mid+,ri,wh<<|);
}
inline void pd(int wh){
    if(col[wh]){
      d[wh<<]+=col[wh];
      col[wh<<]+=col[wh];
      d[wh<<|]+=col[wh];
      col[wh<<|]+=col[wh];
      col[wh]=;
    }
}
inline void up(int wh){
    d[wh]=max(d[wh<<],d[wh<<|]);
    sum[wh]=(d[wh<<]==d[wh]?sum[wh<<]:)+(d[wh<<|]==d[wh]?sum[wh<<|]:);
}
inline void update(int le,int ri,int wh,int x,int y){
    if(le>=x&&ri<=y){
      col[wh]++;
      d[wh]++;
      return;
    }
    pd(wh);
    int mid=(le+ri)>>;
    if(mid>=x)update(le,mid,wh<<,x,y);
    if(mid<y)update(mid+,ri,wh<<|,x,y);
    up(wh);
}
inline void que(int le,int ri,int wh,int x,int y){
    if(le>=x&&ri<=y){
      if(d[wh]>Max)Max=d[wh],cnt=sum[wh];
        else if(d[wh]==Max)cnt+=sum[wh];
      return;
    }
    pd(wh);
    int mid=(le+ri)>>;
    if(mid>=x)que(le,mid,wh<<,x,y);
    if(mid<y)que(mid+,ri,wh<<|,x,y);
    up(wh);
}
inline void add(int x){
    for(int i=;i<low[x].size();i++){
      update(,n,,,low[x][i]);
      if(low[x][i]>=L)num+=((++du[x])==)+((++du[low[x][i]])==);
    }
}
inline void deal(int x){
    for(int i=;i<high[x].size();i++){
      if(high[x][i]<=R)num-=((--du[x])==)+((--du[high[x][i]])==);
    }
}
int main(){
    int i,j,k;
    scanf("%d",&n);
    for(i=;i<n;i++){
      int x,y;
      scanf("%d%d",&x,&y);
      if(x>y)swap(x,y);
      high[x].push_back(y);
      low[y].push_back(x);
    }
    L=;
    build(,n,);
    for(i=;i<=n;i++){
      R=i;
      add(i);
      while(num)deal(L),L++;
      Max=;
      que(,n,,L,i);
      if(Max==i)Ans+=1ll*cnt;
    }
    cout<<Ans;
    return ;
}