hdu5542(树状数组优化dp)

很容易想到n^3的dp,,但是要超时,用二维树状数组优化一下枚举上一个点的第三重循环就好了。

#include
#include
#include
using namespace std;
const int maxn = 1005;
const int mod = 1000000007;
typedef long long ll;
int n;
int num[maxn];
int tmp[maxn];
ll tree[maxn][maxn];
inline int lowbit(int i)
{
    return i&(-i);
}
inline void add(int i,int j,int x)
{
    while(i<=n)
    {
        tree[i][j] += x%mod;
        tree[i][j]%=mod;
        i += lowbit(i);
    }
}
inline ll getsum(int i,int j)
{
    ll sum = 0;
    while(i)
    {
        sum += tree[i][j]%mod;
        sum %= mod;
        i -= lowbit(i);
    }
    return sum;
}
int main()
{
    int cases,m,t=1;
    scanf("%d",&cases);
    while(cases--)
    {
        memset(tree,0,sizeof(tree));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&num[i]),tmp[i] = num[i];
        sort(tmp+1,tmp+1+n);
        for(int i=1;i<=n;i++) num[i] = lower_bound(tmp+1,tmp+1+n,num[i])-tmp;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=min(i,m);j++)
            {
                if(j==1) add(num[i],1,1);
                else
                {
                    ll temp = getsum(num[i]-1,j-1);
                    add(num[i],j,temp);
                }
//                for(int k=1;k


你可能感兴趣的:(dp)