Codeforces 597C Subsequences dp + 树状数组

题意

  • 给你n的一个排列,问你包含k个数的子序列个数有多少。
  • n <= 10^5,k <= 11

思路

  • dp
  • 状态dp[i][j] 表示前i个数,包含k个数的子序列个数有多少
  • f[i][j]表示前i个数,以a[i]结尾的,包含k个数的子序列有多少
  • dp[i][j] = dp[i-1][j] + f[i][j] 这个转移很容易想到
  • 具体f[i][j]怎么转移,稍微有点麻烦
  • f[i][j] = sum f[k][j-1] 其中1<=k < i 且 a[k] < a[i]
  • 这个转移也不难理解,但是转移的复杂度是O(n)的了,我们用树状数组来解决问题
  • 由于1<=a[i] <= n,所以我们可以以a[i]作为下标建立k个树状数组,第j个树a[i]的位置存f[i][j],当我们更新f[i]时,此时我们只需查询1~a[i]的前缀和即可,最后再用f[i]更新树的a[i]节点即可

实现

#include <cstring>
#include <algorithm>
#include <cstdio>
#include <iostream>
using namespace std;
const int maxn = 100005;
int n,k;
typedef long long ll;
int a[maxn];
ll dp[maxn][12];
ll f[maxn][12];

ll tree[maxn][12];
inline int lowbit(int x)
{
    return x&(-x);
}

void update(int i,ll x,int j)
{
    while(i<=n)
    {
        tree[i][j] += x;
        i += lowbit(i);
    }
}

ll query(int i,int j)
{
    ll ret = 0;
    while(i > 0)
    {
        ret += tree[i][j];
        i -= lowbit(i);
    }
    return ret;
}

int main(){
    cin>>n>>k;
    k++;
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for (int i=1;i<=n;i++)
    {
        f[i][1] = 1;
        dp[i][1] = i;
    }   
    update(a[1],1,1);
    for (int i=2;i<=n;i++){
        for (int j=min(i,k);j>1;j--){
            f[i][j] = query(a[i],j-1);
            update(a[i],f[i][j],j);
        }
        update(a[i],1,1);
    }
    for (int i=2;i<=n;i++){
        for (int j=2;j<=min(i,k);j++){
            dp[i][j] = dp[i-1][j] + f[i][j];
        }
    }
    cout << dp[n][k] << "\n";
    return 0;
}
树状数组(Fenwick Tree)是一种用于高效处理区间和查询的数据结构,常用于解一维数组的前缀和、区间更新和查询等问题。 在 Codeforces 上,树状数组常被用来解决一些与区间和查询有关的问题。它可以在 O(logn) 的时间内完成单点更新和查询,以及区间求和等操作。 下面是一个简单的示例代码,展示了如何实现一个基本的树状数组: ```cpp #include <iostream> #include <vector> using namespace std; // 获取最低位的 1 int getLowbit(int x) { return x & -x; } // 树状数组的单点更新操作 void update(vector<int>& fenwick, int index, int delta) { while (index < fenwick.size()) { fenwick[index] += delta; index += getLowbit(index); } } // 树状数组的前缀和查询操作 int query(vector<int>& fenwick, int index) { int sum = 0; while (index > 0) { sum += fenwick[index]; index -= getLowbit(index); } return sum; } int main() { int n; cin >> n; vector<int> fenwick(n + 1, 0); // 初始化树状数组 for (int i = 1; i <= n; i++) { int val; cin >> val; update(fenwick, i, val); } // 进行查询操作 int q; cin >> q; while (q--) { int type; cin >> type; if (type == 1) { int index, delta; cin >> index >> delta; update(fenwick, index, delta); } else if (type == 2) { int l, r; cin >> l >> r; int sum = query(fenwick, r) - query(fenwick, l - 1); cout << sum << endl; } } return 0; } ``` 在这个示例中,我们使用了一个长度为 n 的数组 `fenwick` 来表示树状数组。`update` 函数用于更新树状数组中的某个元素,`query` 函数用于查询树状数组中某个区间的和。 你可以根据具体问题的要求进行相应的修改和扩展。希望对你有所帮助!如果有任何疑问,请随时提问。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值