ZOJ 3606 Lazy Salesgirl ( 线段树 + 思路 )

卖切糕的小女孩

http://www.cnblogs.com/wuyiqi/archive/2012/04/28/2474672.html

#include <cstdio>
#include <cstring>
#include <algorithm>

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1
#define lc rt << 1
#define rc rt << 1 | 1
#define LL long long int

using namespace std;

const int MAXN = ;
const double INF = 1e100;
const double eps = 1e-;

struct node
{
    int p;
    int t;
    int id;
};

int N;
int cnt[MAXN << ];
LL sum[][MAXN << ];
node D1[MAXN], D2[MAXN];

bool cmp( const node& a, const node& b )
{
    return a.t < b.t;
}

void build( int l, int r, int rt )
{
    cnt[rt]=;
    for ( int i = ; i < ; ++i ) sum[i][rt] = ;
    if ( l == r ) return;
    int m = ( l + r ) >> ;
    build( lson );
    build( rson );
    return;
}

void PushUp( int rt )
{
    cnt[rt] = cnt[lc] + cnt[rc];
    for ( int i = ; i < ; ++i )
        sum[i][rt] = sum[i][lc] + sum[ (i+cnt[lc])% ][rc];
    return;
}

void update( int L, int l, int r, int rt )
{
    if ( l == r && L == l )
    {
        ++cnt[rt];
        for ( int i = ; i < ; ++i )
            sum[i][rt] += (LL)D1[L].p * (i + );
        return;
    }

    int m = ( l + r ) >> ;
    if ( L <= m ) update( L, lson );
    else update( L, rson );
    PushUp( rt );
    return;
}

void init()
{
    scanf( "%d", &N );
    build( , N,  );
    for ( int i = ; i <= N; ++i )
        scanf( "%d", &D1[i].p );
    for ( int i = ; i <= N; ++i )
        scanf( "%d", &D1[i].t );

    sort( D1 + , D1 + N + , cmp );

    D2[].t = ;
    for ( int i = ; i <= N; ++i )
    {
        D2[i].id = i;
        D2[i].t = D1[i].t - D1[i - ].t;
    }
    sort( D2 + , D2 + N + , cmp );
    return;
}

void solved()
{
    double minW;
    double maxAve = 0.0;
    int i = , j = ;
    while ( i <= N )
    {
        while ( j <= N && D2[i].t == D2[j].t )
        {
            update( D2[j].id, , N,  );
            ++j;
        }

        double tmpW = (double)D2[i].t;
        double tmpA = (double)sum[][]/cnt[];
        //printf( "tmp: %f %f\n", tmpW, tmpA );

        if ( tmpA > maxAve )
        {
            maxAve = tmpA;
            minW = tmpW;
        }
        else if ( tmpA == maxAve )
        {
            if ( minW > tmpW )
                minW = tmpW;
        }
        i = j;
    }
    printf( "%.6f %.6f\n", minW, maxAve );
    return;
}

int main()
{
    int T;
    scanf( "%d", &T );
    while (T--)
    {
        init();
        solved();
    }
    return ;
}