一个很简单的问题,暴力枚举k,然后每次的复杂度位O(n/k);这样很明显的nlog(n);
至于每次在数组A 上求一个L , R 使得sum(L , R) * MIN(L ,R) 最大,只需维护每个点的最大延伸区间,可以用栈边走变维护,或者直接保留成数组递推。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define rep(i,n) for(int i =0;i<(int)n;i++)
#define rep1(i,x,y) for(int i=(int)x;i<=(int)y;i++)
const int maxn = 3e5 + 100;
int L[maxn],R[maxn];
ll sum[maxn];
void cal(int n ,int* a)
{
L[1] = 0;
for(int i =2 ; i<=n ; i++)
{
int now = i - 1;
while(now && a[now] >= a[i])
{
now = L[now];
}
L[i] = now;
}
R[n] = n + 1;
for(int i = n - 1 ; i>=1; i--)
{
int now = i + 1;
while(now != n+1 && a[now] >= a[i])
{
now = R[now];
}
R[i] = now;
}
sum[0] = 0;
for(int i = 1 ; i<=n ; i++) sum[i] = sum[i - 1] + a[i];
}
int a[maxn] ,b[maxn];
int n;
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
rep1(i , 1 , n) scanf("%d",&a[i]);
ll ans = 0;
for(int k = 1 ; k<=n ; k++)
{
int lim = n / k;
int sqrtk = (int) sqrt(k);
int m = 0;
for(int j = k ; j<= n ; j+=k) {
b[++m] = a[j];
}
cal(m, b);
for(int p = 1 ; p<= m; p++)
{
int l = L[p] + 1;
int r = R[p] - 1;
ans = max((sum[r] - sum[l - 1])* b[p] * sqrtk , ans);
}
}
printf("%I64d\n",ans);
}
return 0;
}