P1637 三元上升子序列(DP+離散化權值樹狀數組)
思路:離散化轉權值樹狀數組。
顯然可以設爲長度爲以結尾的子序列的個數。
有轉移方程:
顯然暴力時間複雜度 :
因爲,但考慮離散化,然後轉權值線段樹儲存。
先初始化一元上升子序列,然後從前往後遍歷,
有轉移方程:
再更新。
時間複雜度:,是幾元上升子序列。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=3e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a) memset(a,0,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first
#define se second
int n,m;
ll a[N],b[N],tr[N];
ll dp[4][N];
#define lowbit(x) x&(-x)
void update(int x,int k){
while(x<=m){
tr[x]+=k;
x+=lowbit(x);
}
}
ll query(int x){
ll ans=0;
while(x){
ans+=tr[x];
x-=lowbit(x);
}
return ans;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]),b[i]=a[i];
dp[1][i]=1;
}
sort(b+1,b+n+1);
m=unique(b+1,b+n+1)-b-1;
for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+m+1,a[i])-b;
for(int i=2;i<=3;i++){
mst(tr);
for(int j=1;j<=n;j++)
{
dp[i][j]+=query(a[j]-1);
update(a[j],dp[i-1][j]);
}
}
ll ans=0;
for(int i=1;i<=n;i++) ans+=dp[3][i];
printf("%lld\n",ans);
return 0;
}