三元上升子序列(洛谷p1637)(树状数组/线段树)(扩展k元上升子序列)

题目地址

预处理l[i],r[i]表示左边比它小的和右边比它大的的个数。
显然，可以用树状数组或者线段树随着枚举，更新区间的值，然后查询一个区间的和，即可以得到l[],r[]。最后ans += l[i]*r[i].

线段树

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 3e4 + 10;
typedef long long ll;

int n,a[N];
vector<int> b;

#define l(x) t[x].l
#define r(x) t[x].r
#define sum(x) t[x].sum

struct SegmentTree{
	int l,r,sum;
}t[N*4];

int l[N],r[N];
int find(int x){
	return lower_bound(b.begin(),b.end(),x)-b.begin();
}

void push_up(int p){
	sum(p) = sum(p*2) + sum(p*2+1);
}

void build(int p,int l,int r){
	l(p) = l,r(p) = r;
	sum(p) = 0;
	if(l == r) return;
	int mid = (l + r) >> 1;
	build(p*2,l,mid);build(p*2+1,mid+1,r);
}

void update(int p,int x,int v){
	if(l(p) == r(p)){
		sum(p) += v;
		return ;
	}
	int mid = (l(p) + r(p)) >> 1;
	if(x <= mid) update(p*2,x,v);
	else update(p*2+1,x,v);
	push_up(p);
}

int ask(int p,int l,int r){
	if(l <= l(p) && r >= r(p)) return sum(p);
	int val = 0,mid = (l(p) + r(p)) >> 1;
	if(l <= mid) val += ask(p*2,l,r);
	if(r > mid) val += ask(p*2+1,l,r);
	return val;
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i],b.push_back(a[i]);
    sort(b.begin(),b.end());
    b.erase(unique(b.begin(),b.end()),b.end());
    build(1,1,n);
    for(int i=1;i<=n;i++) a[i] = find(a[i]) + 1;
    for(int i=1;i<=n;i++){
    	update(1,a[i],1);
    	if(a[i] != 1) l[i] = ask(1,1,a[i]-1);
	}
	build(1,1,n);//可以直接清空sum = 0; 
	for(int i=n;i>=1;i--){
		update(1,a[i],1);
		r[i] = ask(1,a[i]+1,n);
	}
	ll ans = 0;
	for(int i=1;i<=n;i++) ans += (l[i] * r[i]);
    cout<<ans<<endl;
    return 0;
}

树状数组

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int N = 3e4 + 10;
typedef long long ll;

int n,a[N],c[N];
int l[N],r[N];
vector<int> b;

int find(int x){
	return lower_bound(b.begin(),b.end(),x)-b.begin();
}

#define lowbit(x) (x & -x)
void update(int x,int v){
	for(int i=x;i<=n;i+=lowbit(i)) c[i] += v;
}

ll ask(int x){
	ll res = 0;
	for(int i=x;i;i-=lowbit(i)) res += c[i];
	return res;
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i],b.push_back(a[i]);
    sort(b.begin(),b.end());
    b.erase(unique(b.begin(),b.end()),b.end());
    for(int i=1;i<=n;i++) a[i] = find(a[i]) + 1;
    for(int i=1;i<=n;i++){
    	update(a[i],1);
    	l[i] = ask(a[i]-1);
	}
	memset(c,0,sizeof(c));
	for(int i=n;i>=1;i--){
		update(a[i],1);
		r[i] = ask(n) - ask(a[i]);
	}
	ll ans = 0;
	for(int i=1;i<=n;i++) ans += (l[i] * r[i]);
    cout<<ans<<endl;
    return 0;
}

扩展k元上升子序列。题目地址
此时设一个状态转移方程dp[i][j] 表示长度为i且以a[j]结尾的上升序列的个数。

状态转移dp[i][j] 为 dp[i-1][k] k < j 且a[k] < a[j].的和转移而来。按升序处理，只需处理j前面的就满足第一个条件。 第二个条件，可以用树状数组或者线段树维护数量。
每次转移只需ask(a[j]-1). 之后更新a[j]的值加dp[i-1][j]即可。

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int N = 1e4 + 10;
const int mod = 123456789;
typedef long long ll;

int n,m,a[N],c[N];
ll dp[110][N];
vector<int> b;

int find(int x){
	return lower_bound(b.begin(),b.end(),x)-b.begin();
}

#define lowbit(x) (x & -x)
void update(int x,ll v){
	for(int i=x;i<=n;i+=lowbit(i)) c[i] = (c[i] + v) % mod;
}

ll ask(int x){
	ll res = 0;
	for(int i=x;i;i-=lowbit(i)) res = (res + c[i]) % mod;
	return res;
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    while(cin>>n>>m){
    	b.clear();
    	for(int i=1;i<=n;i++) cin>>a[i],b.push_back(a[i]);
	    sort(b.begin(),b.end());
	    b.erase(unique(b.begin(),b.end()),b.end());
	    for(int i=1;i<=n;i++) a[i] = find(a[i]) + 1,dp[1][i] = 1;
	    for(int i=2;i<=m;i++){
	    	memset(c,0,sizeof(c));
	    	for(int j=1;j<=n;j++){
	    		dp[i][j] = ask(a[j]-1) % mod;
	    		update(a[j],dp[i-1][j]);
			}
		}
		ll ans = 0;
		for(int i=1;i<=n;i++) ans = (ans + dp[m][i]) % mod;
	    cout<<ans<<endl;
	}
    return 0;
}