洛谷P1637 三元上升子序列

洛谷P1637 三元上升子序列

题意：Erwin 最近对一种叫 thair 的东西巨感兴趣。。。在含有 n 个整数的序列$ a_1,a_2,…,a_n $中，三个数被称作‘thair‘当且仅当$i<j<k$ 且 $a_i<a_j<a_k$;求一个序列中 thair 的个数。

输入：

4
2 1 3 4
5
1 2 2 3 4    

输出：

2
7

DP+动态规划+离散化

• 假设$f[i][j]$表示以$a[j]$结尾的长度为 i 的上升子序列的个数。

• 动态转移方程$f[i][j]={\sum }_k^{j}f[i-1][k]$ , $a[k]<a[j]$;

• 要满足两个性质：

  • $k<j$;
  • $a[k]<a[j];$

求和我们可以用树状数组+离散化来维护，

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define sc(n) scanf("%lld",&n)
#define SC(a,b) scanf("%lld%lld",&a,&b)
#define pr(a) printf("%lld",a)
const int N=1000010;
int n,a[1000010],c[1000010],b[N],m,ans,f[4][300010];
//f[i][j]表示以a[j]结尾的长度为i,上升子序列的个数
int lowbit(int x){
    return x&-x;//返回二进制最后一位1
}

void updata(int i,int k){//在i的位置上加上k
    for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){//求a[1]-a[i]的和
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}


signed main(){
	sc(n);
	for(int i=1;i<=n;i++) sc(a[i]),b[i]=a[i];//离散化操作
	sort(b+1,b+1+n);
	int cnt=unique(b+1,b+1+n)-(b+1);
	for(int i=1;i<=n;i++){
		f[1][i]=1;//长度为1的个数初始化为1
		a[i]=lower_bound(b+1,b+1+n,a[i])-b;//离散化
	}
	for(int i=2;i<=3;i++){
		memset(c,0,sizeof(c));
		for(int j=1;j<=n;j++){
			f[i][j]=getsum(a[j]-1);//动态转移方程
			updata(a[j],f[i-1][j]);
		}
	}
	for(int i=1;i<=n;i++){
		ans+=f[3][i];
	}
	pr(ans);
	return 0;
}