[HDU 5542] The Battle of Chibi

[题目链接]

http://acm.hdu.edu.cn/showproblem.php?pid=5542

[算法]

树状数组优化DP

[代码]

#include<bits/stdc++.h>
using namespace std;
#define MAXN 1010
const int P = 1e9 + ;

int i,j,T,TC,n,m,len,ans;
int a[MAXN],tmp[MAXN],rk[MAXN];
int f[MAXN][MAXN];

class BinaryIndexedTree
{
    private :
        int c[MAXN];
    public :
        inline int lowbit(int x)
        {
            return x & (-x);
        }
        inline void clear()
        {
            memset(c,,sizeof(c));
        }
        inline void modify(int pos,int val)
        {
            int i;
            for (i = pos; i <= len; i += lowbit(i)) c[i] =(c[i] + val) % P;
        }
        inline int query(int pos)
        {
            int i;
            int ret = ;
            for (i = pos; i; i -= lowbit(i)) ret = (ret + c[i]) % P;
            return ret;
        }
} BIT;

int main()
{

    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&n,&m);
        for (i = ; i <= n; i++)
        {
            scanf("%d",&a[i]);
            tmp[i] = a[i];
        }
        sort(tmp+,tmp+n+);
        len = unique(tmp+,tmp+n+) - tmp - ;
        for (i = ; i <= n; i++) rk[i] = lower_bound(tmp+,tmp+len+,a[i]) - tmp;
        for (i = ; i <= n; i++) f[][i] = ;
        for (i = ; i <= m; i++)
        {
            BIT.clear();
            for (j = ; j <= n; j++)
            {
                f[i][j] = BIT.query(rk[j] - );
                BIT.modify(rk[j],f[i-][j]);
            }
        }
        ans = ;
        for (i = ; i <= n; i++) ans = (ans + f[m][i]) % P;
        printf("Case #%d: %d\n",++TC,ans);
    }

    return ;
}