2015 CCPC C题 【树状数组优化dp】

题目链接 点我

题意：给出n个数组成的序列，让你按顺序任选m个数，问得到的上升子序列个数 % 1000000007。

思路：较简单的dp，状态转移方程也好想。

用dp[i][j]表示以a[i]结尾的长度为j的上升子序列个数。

状态转移方程dp[i][j] = sigma(dp[k][j-1]) 其中(1 <= k < i && a[k] < a[i])。

主要是O(n^3)的时间复杂度过高，需要优化。我们考虑a[k] < a[i] 且 1 <= k < i，如果能够想到逆序对就成功了一半。 我们只需要用树状数组处理出a[i]前面比它小的数即可，这里我用了二维树状数组来维护。

时间复杂度为O(nmlog(n))

例如：

4 3

1 2 3 4

维护过程如下：横行为结尾元素编号，纵行为子序列长度，维护到最后，sum(4, 3)就是结果。

	0	1	2	3	4
0	0	0	0	0	0
1	0	1	1	1	1
2	0	0	1	2	3
3	0	0	0	1	3

AC代码：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <map>
#include <string>
#include <vector>
#define lson o<<1|1, l, mid
#define rson o<<1, mid+1, r
#define ll o<<1
#define rr o<<1|1
#define INF 0x3f3f3f3f
#define eps 1e-8
#define debug printf("1\n")
#define MAXN 1010
#define MAXM 100000
#define LL long long
#define CLR(a, b) memset(a, (b), sizeof(a))
#define W(a) while(a--)
#define Ri(a) scanf("%d", &a)
#define Ri2(a, b) scanf("%d%d", &a, &b)
#define Pi(a) printf("%d\n", (a))
#define Pi2(a, b) printf("%d %d\n", a, b)
#define Rl(a) scanf("%lld", &a)
#define Rl2(a, b) scanf("%lld%lld", &a, &b)
#define Pl(a) printf("%lld\n", (a))
#define Pl2(a, b) printf("%lld %lld\n", a, b)
#define Rs(a) scanf("%s", a)
#define Rs2(a, b) scanf("%s%s", a, b)
#define Ps(a) printf("%s\n", (a))
#define Ps2(a, b) printf("%s %s\n", a, b)
#define FOR(i, l, r) for(int i = (l); i <= (r); i++)
#define FOR1(i, l, r) for(int i = (l); i < (r); i++)
#define MOD 1000000007
using namespace std;
int a[MAXN], dp[MAXN][MAXN];
int n, m;
int lowbit(int x){
    return x & (-x);
}
void update(int x, int d, int index)
{
    while(x <= n)
    {
        dp[x][index] += d;
        dp[x][index] %= MOD;
        x += lowbit(x);
    }
}
int sum(int x, int index)
{
    int s = 0;
    while(x > 0)
    {
        s += dp[x][index];
        s %= MOD;
        x -= lowbit(x);
    }
    return s;
}
struct MES{
    int val, id;
};
MES num[MAXN];
bool cmp(MES a, MES b){
    if(a.val != b.val)
        return a.val < b.val;
    else
        return a.id > b.id;
}
int main()
{
    int t, kcase = 1;
    Ri(t);
    W(t)
    {
        Ri(n), Ri(m);
        FOR(i, 1, n)
        {
            Ri(num[i].val);
            num[i].id = i;
        }
        sort(num+1, num+n+1, cmp);
        CLR(dp, 0);
        FOR(i, 1, n)
        {
            //dp[i][1] = 1;
            FOR(j, 1, min(i, m))
            {
                if(j == 1)
                    update(num[i].id, 1, j);
                else
                {
                    int sub = sum(num[i].id-1, j-1);
                    update(num[i].id, sub, j);
                }
            }
        }
        printf("Case #%d: ", kcase++);
        Pi(sum(n, m));
    }
    return 0;
}