[atAGC053E]More Peaks More Fun

假设已经确定顺序(且不妨假设$a_{i}<b_{i}$),考虑如何判定是否合法——

显然$a_{i}$不能为峰且峰不能相邻,因此峰数的上限是$n-1$

结论:合法当且仅当存在$k\in [0,n]$使得$\forall 1\le i<k,a_{i+1}<b_{i}$且$\forall k+2\le i\le n,a_{i-1}<b_{i}$

为了描述方便,称第$i$组先加入$a_{i}$再加入$b_{i}$为顺序,反之为逆序

充分性:若满足此条件,将$[1,k]$顺序加入且$(k,n]$逆序加入,根据条件不难得到$\forall i\not\in \{k,k+1\},b_{i}$均是峰,同时在$b_{k}$和$b_{k+1}$中较大的一项也总是峰(特别的,当$k\in \{0,n\}$时之前已经有$n-1$个峰)

必要性:对于$i\in [1,n)$,若$i$是顺序且$i+1$是逆序,那么$b_{i}$和$b_{i+1}$中至少有一个不为峰

特别的,若1是逆序或$n$是顺序,那么$b_{1}$或$b_{n}$不为峰(可以理解为0是顺序且$n+1$是逆序)

结合两者,即存在$k\in [0,n]$,满足$[1,k]$为顺序$,(k,n]$为逆序且$\forall i\not\in\{k,k+1\},b_{i}$均是峰,也即条件

进一步的,注意到$k$仅是确定范围,因此总可以取$k=\min_{1\le i\le n,a_{i+1}>b_{i}}i$(约定$a_{n+1}=\infty$),并判定后者是否成立

回到原问题,考虑初始将其按$b_{i}$从小到大排序,并记$U_{i}=\sum_{i<j,a_{j}<b_{i}}1$

对于$k=n$的情况,考虑从后往前依次加入,第$i$个可行的位置有$U_{i}+1$个(注意$b_{i}$的单调性),因此方案数为$\prod_{i=1}^{n}(U_{i}+1)$

对于$k<n$的情况,枚举$k$和$k+1$对应的编号$x$和$y$(保证$a_{y}>b_{x}$,进而显然$x<y$),将$(k,n]$翻转,问题即可以看作构造两个分别以$x$和$y$结尾的序列(条件类似$k=n$),类似地方案数为$\prod_{i=1}^{x-1}U_{i}\prod_{i=x+1}^{y-1}(U_{i}+1)\prod_{i=y+1}^{n}(U_{i}+2)$(考虑结尾能否选)

关于这个问题,用一个树状数组维护即可,时间复杂度为$o(n\log n)$,可以通过

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 200005
 4 #define mod 1000000007
 5 #define ll long long
 6 struct Data{
 7     int a,b;
 8     bool operator < (const Data &k)const{
 9         return b<k.b;
10     }
11 }a[N];
12 int n,ans,inv[N],U[N],pre[N],suf1[N],suf2[N],f[N<<1];
13 int lowbit(int k){
14     return (k&(-k));
15 }
16 void update(int k,int x){
17     while (k<=(n<<1)){
18         f[k]=(f[k]+x)%mod;
19         k+=lowbit(k);
20     }
21 }
22 int query(int k){
23     int ans=0;
24     while (k){
25         ans=(ans+f[k])%mod;
26         k-=lowbit(k);
27     }
28     return ans;
29 }
30 int main(){
31     inv[0]=inv[1]=1;
32     for(int i=2;i<N;i++)inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod;
33     scanf("%d",&n);
34     for(int i=1;i<=n;i++){
35         scanf("%d%d",&a[i].a,&a[i].b);
36         if (a[i].a>a[i].b)swap(a[i].a,a[i].b);
37     }
38     sort(a+1,a+n+1);
39     for(int i=n;i;i--){
40         U[i]=query(a[i].b);
41         update(a[i].a,1);
42     }
43     pre[0]=suf1[n+1]=suf2[n+1]=1;
44     for(int i=1;i<=n;i++)pre[i]=(ll)pre[i-1]*U[i]%mod;
45     for(int i=n;i;i--)suf1[i]=(ll)suf1[i+1]*(U[i]+1)%mod;
46     for(int i=n;i;i--)suf2[i]=(ll)suf2[i+1]*(U[i]+2)%mod*inv[U[i]+1]%mod;
47     memset(f,0,sizeof(f));
48     ans=suf1[1];
49     for(int i=n;i;i--){
50         ans=(ans+(ll)pre[i-1]*suf1[i+1]%mod*(query(n<<1)-query(a[i].b)+mod))%mod;
51         update(a[i].a,(ll)suf2[i+1]*inv[U[i]+1]%mod);
52     }
53     printf("%d\n",ans);
54     return 0;
55 }
View Code

 

posted @ 2021-12-24 15:52  PYWBKTDA  阅读(92)  评论(0编辑  收藏  举报