POJ 3378

题目链接

查找长度为5的上升序列总数

用的树状数组+高精度

用树状数组求在i前面比i小的数有几个

用的4个树状数组，A[i][j]表示长度为i的以j为结尾的个数，A[i][j]=A[i-1][1....flag[n]]

对数组下标不能为0一直不太理解，会发生死循环

 

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
#define N 50005
#define MOD 10000
struct e{
    int w,id;
    bool operator <(const e &v) const
    {
        return w<v.w;
    }
}a[N];
__int64 A[4][N];
int flag[N],ans[10],n;
int lowbit(int x){ return x&-x; }
void update(int x,int y,__int64 w)
{
    while(y<n)
    {
        A[x][y]+=w;
        y+=lowbit(y);
    }

}
__int64 getsum(int x,int y)
{
    __int64 sum=0;
    while(y>0)
    {
        sum+=A[x][y];
        y-=lowbit(y);
    }
    return sum;
}
int ans_sum(__int64 w)  //高精度取模
{
    int i;
    i=0;
    while(w>0)
    {
        ans[i]+=w%MOD;
        if(ans[i]>=MOD)
        {
            ans[i]-=MOD;
            ans[i+1]++;
        }
        w/=MOD;
        i++;
    }
}
int main()
{
    int i,j;
    while(scanf("%d",&n)!=EOF)
    {
      //  memset(arr,0,sizeof(arr));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i].w);
            a[i].id=i;
        }
        sort(a,a+n);
        memset(ans,0,sizeof(ans));
        memset(A,0,sizeof(A));
        int mm=0;
        flag[a[1].id]=++mm;
        for(i=2;i<=n;i++)
        {
            if(a[i-1].w<a[i].w)
                flag[a[i].id]=++mm;
            else
                flag[a[i].id]=mm;
        }
        for(i=1;i<=n;i++)
        {
            __int64 x=1;
            for(j=0;j<4;j++)
            {
                update(j,flag[i],x); //x为长度为j-1的总数
                x=getsum(j,flag[i]-1);  
            }
            ans_sum(x); //对满足条件的进行求和
        }
        int t=0;
        for(i=9;i>=0;i--)
        {
            if(!t&&ans[i]>0)
            {
                printf("%d",ans[i]);
                t=1;
            }
            else if(t)
            printf("%04d",ans[i]);
        }
        if(!t) printf("0");
        printf("\n");
    }
    return 0;
}