HDU 5358 First One

这题你必须严格地写出 O(nlog(n)) 的时间复杂度，我们尝试了下 O(nlog(n)2) ，结果不行，而且卡常数卡得特别死。

首先枚举 log(SUM) 的值，从1~33，每次固定左端点 i ，然后用尺取法枚举出符合位数的 SUM[l,r] ，这里面的sum的对数值相同，可以直接统一计算。
所以总体复杂度为 O(nlog(n))

我写的尺取法是左闭右闭的，据说左闭右开的区间似乎更快一些。

// whn6325689
// Mr.Phoebe
// http://blog.csdn.net/u013007900
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <climits>
#include <complex>
#include <fstream>
#include <cassert>
#include <cstdio>
#include <bitset>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <ctime>
#include <set>
#include <map>
#include <cmath>
#include <functional>
#include <numeric>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define eps 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LLINF 1LL<<62
#define speed std::ios::sync_with_stdio(false);

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef complex<ld> point;
typedef pair<int, int> pii;
typedef pair<pii, int> piii;
typedef vector<int> vi;

#define CLR(x,y) memset(x,y,sizeof(x))
#define CPY(x,y) memcpy(x,y,sizeof(x))
#define clr(a,x,size) memset(a,x,sizeof(a[0])*(size))
#define cpy(a,x,size) memcpy(a,x,sizeof(a[0])*(size))

#define mp(x,y) make_pair(x,y)
#define pb(x) push_back(x)
#define lowbit(x) (x&(-x))

#define MID(x,y) (x+((y-x)>>1))
#define ls (idx<<1)
#define rs (idx<<1|1)
#define lson ls,l,mid
#define rson rs,mid+1,r
#define root 1,1,n

template<class T>
inline bool read(T &n)
{
    T x = 0, tmp = 1;
    char c = getchar();
    while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
    if(c == EOF) return false;
    if(c == '-') c = getchar(), tmp = -1;
    while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
    n = x*tmp;
    return true;
}
template <class T>
inline void write(T n)
{
    if(n < 0)
    {
        putchar('-');
        n = -n;
    }
    int len = 0,data[20];
    while(n)
    {
        data[len++] = n%10;
        n /= 10;
    }
    if(!len) data[len++] = 0;
    while(len--) putchar(data[len]+48);
}
//-----------------------------------

const int MAXN=100010;

int n;
ll sum[MAXN],a[MAXN];
ll pow2[40];

int main()
{
    pow2[0]=1;
    for(int i=1;i<40;i++)
        pow2[i]=pow2[i-1]<<1;
    int T;
    read(T);
    while(T--)
    {
        read(n);
        sum[0]=0;
        for(int i=1;i<=n;i++)
            read(a[i]),sum[i]=sum[i-1]+a[i];
        int be,en;
        ll ans=0,temp;
        for(int i=1;i<=35;i++)
        {
            if(pow2[i]>sum[n])break;
            int l=1,r=0;temp=0;
            for(int j=1;j<=n;j++)
            {
                while(l<=n && sum[l]-sum[j-1]<pow2[i])l++;
                while(r+1<=n && sum[r+1]-sum[j-1]<pow2[i+1])r++;
                if(l<=r)
                    temp+=(r+l*1LL)*(r-l+1LL)/2LL+j*(r-l+1LL);
            }
            ans+=temp*i;
        }
        for(int i=1;i<=n;i++)
            ans+=(n+i)*1LL*(n-i+1)/2+i*1LL*(n-i+1);
        write(ans),putchar('\n');
    }
    return 0;
}