HDU 3030 Increasing Speed Limits

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

本来想的是DP的算法，但是DP的话明显要超时的。

然后文哥告诉我的用树状数组的方法，我一直想怎么用树状数组来表示。。

看文哥模拟，然后知道。。。

想将数组离散化，然后记录第i个位置为结尾的序列的个数（就是DP的思想了）。。

并且每次都要更新以 第 i 个元素为结尾的序列的当前的个数，这里用到了树状数组更新。

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 500005;
const int mod = 1000000007;
long long  ans[maxn],has[maxn],temp[maxn],x,y,z,n,m;

int lowbit(int i)
{
    return i & (-i);
}

long long getSum(int i)
{
    long long sum=0;
    while(i)
    {
        sum+=temp[i];
        sum%=mod;
        i-=lowbit(i);
    }
    return sum;
}

void modfiy(int i,int c)
{
    while(i<maxn)
    {
        temp[i]+=c;
        i+=lowbit(i);
    }
}

void make()
{
    int i,j;
    j=1;
    if(m>=n)
    {
        for(i=0;i<n;i++)
            ans[j++]=has[i];
        
    }
    else
    {
        for(i=0;i<n;i++)
        {
            ans[j++]=has[i%m];
            has[i%m]=(x*has[i%m]+y*(i+1))%z;
        }
    }
    for(i=1;i<=n;i++)
        temp[i]=ans[i];
    sort(temp+1,temp+n+1);
    has[1]=temp[1];
    for(i=2,j=2;i<=n;i++)
        if(temp[i]!=temp[i-1])
            has[j++]=temp[i];
    m=j;
}

int search(long long num)
{
    int i,j,mid;
    i=1;    j=m-1;
    while(i<=j)
    {
        mid=(i+j)/2;
        if(num==has[mid])
            return mid;
        if(num>has[mid])
            i=mid+1;
        else
            j=mid-1;
            
    }
    return mid;
}


int main()
{
    int t,i,j,l,ch;
    cin>>t;
    for(l=1;l<=t;l++)
    {
        cin>>n>>m>>x>>y>>z;
        for(i=0;i<m;i++)
            cin>>has[i];
        make();
        for(i=0;i<maxn;i++)
            temp[i]=0;
        for(i=1;i<=n;i++)
        {
            ch=search(ans[i]);
            ans[i]=getSum(ch-1)+1;
            modfiy(ch,ans[i]);
        }
        long long sum=0;
        cout<<"Case #"<<l<<": ";
        for(i=1;i<=n;i++)
            sum=(sum+ans[i])%mod;
        cout<<sum<<endl;
    }
    return 0;
}