ZCMU 1708 树状数组优化dp

1708: 上升子序列

Time Limit: 1 Sec   Memory Limit: 128 MB

Description

有一个长度为n的数组，要你求出最长上升子序列？？？然而事实并没有那么难。infact,有一个长度为n的数组a，求出长度为m的上升子序列的个数。

Input

多组测试数据

第一行输入一个n，m（m<=n<=1000）

第二行输入数组a。1<=ai<=1000

Output

输出长度为m上升子序列的个数，结果对10007取模。

Sample Input

5 11 1 1 1 15 21 2 3 4 55 21 1 1 1 1

Sample Output

5100

树状数组优化dp

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<string>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
#define ll long long
#define cl(a,b) memset(a,b,sizeof(a))
#define N 50005
#define MAXN 100005
#define mod 10007
#define fINF -201510005
int dp[1005][1005];
int c[1005][1005];
int a[1005],n,m,num;
inline int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int j,int val)
{
    while(j<num)
    {
        c[i][j]+=val;
        if(c[i][j]>=mod)c[i][j]%=mod;
        j+=lowbit(j);
    }
    return;
}
int sum(int i,int j)
{
    int re=0;
    while(j)
    {
        re+=c[i][j];
        if(re>=mod)re%=mod;
        j-=lowbit(j);
    }
    return re;
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        cl(dp,0);
        cl(c,0);
        set<int>s;
        map<int,int>mp;
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            s.insert(a[i]);
        }
        num=1;
        for(set<int>::iterator it=s.begin();it!=s.end();++it)
            mp[*it]=num++;
        for(int i=0;i<n;i++)
            for(int j=1;j<=m;j++)
            {
                if(j==1)
                    dp[i][j]=1;
                else
                {
                    dp[i][j]+=sum(j-1,mp[a[i]]-1);
                    dp[i][j]%=mod;
                }
                add(j,mp[a[i]],dp[i][j]);
            }
        int ans=0;
        for(int i=0;i<n;i++)
        {
            ans+=dp[i][m];
            ans%=mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}