hdu--2227--dp<线段树/树状数组>优化

一开始写了发 直接O(n^2)的 tle = =

后来 看了别人的 原来可以这样优化

又学到了。

本来想多说点关于这个优化细节的  有点烦 不想说什么了 自己看下代码吧    

很多时候 dp都需要优化= =

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

typedef __int64 LL;
int n , cnt;
const int mod = 1000000007;
const int size = 100010;
LL tree[size];
int a[size] , b[size];

int lowBit( int x )
{
    return x & -x;
}

void update( int x , LL var )
{
    while( x<=cnt-1 )
    {
        tree[x] += var;
        tree[x] %= mod;
        x += lowBit( x );
    }
}

LL getSum( int x )
{
    LL sum = 0;
    while( x )
    {
        sum += tree[x];
        sum %= mod;
        x -= lowBit( x );
    }
    return sum;
}

LL solve( )
{
    LL sum = 0;
    int Id;
    LL var;
    for( int i = 1 ; i<=n ; i++ )
    {
        Id = lower_bound( b+1 , b+cnt , a[i] ) - b;
        var = getSum( Id ) + 1;
        sum += var;
        sum %= mod;
        update( Id , var );
    }
    return sum;
}

int main()
{
    cin.sync_with_stdio(false);
    int num;
    while( cin >> n )
    {
        memset( tree , 0 , sizeof(tree) );
        for( int i = 1 ; i<=n ; i++ )
        {
            cin >> num;
            a[i] = b[i] = num;
        }
        sort( b+1 , b+n+1 );
        cnt = unique( b+1 , b+n+1 ) - b;
        cout << solve( ) << endl;
    }
    return 0;
}