LibreOJ β Round #2 A. 模拟只会猜题意【前缀和】

A. 模拟只会猜题意

内存限制：256 MiB 时间限制：1000 ms 标准输入输出

题目类型：传统 评测方式：文本比较

转到题库 提交记录 返回比赛

题目描述

给定一个长度为 $n$ 的序列 $A$ 。

定义 f(l,r)=∑i=lrAif(l,r)=\sum_{i=l}^{r} A_{i}f(l,r)=∑​i=l​r​​A​i​​。

询问 $m$ 次，每次询问一个数字 $x$，请求出所有满足 $r-l+1\ge x$ 区间 $[l,r]$ 中最大的 $f(l,r)$。

输入格式

第一行两个数，表示 $n$ 和 $m$ 。
之后 $n$ 个数，表示序列 $A$。
之后 $m$ 行每行一个数 $x$，表示询问 $x$ 。

输出格式

输出 $m$ 行，每行一个答案，表示最大的 $f(l,r)$ 。

样例

样例输入

5 5
1 2 3 4 5
1
2
3
4
5

样例输出

15
15
15
15
15

数据范围与提示

1≤x≤n≤1041 \leq x \leq n \leq 10^{4}1≤x≤n≤10​4​​ ，0≤m≤1050 \leq m \leq 10^{5}0≤m≤10​5​​ ， ∣Ai∣≤104|A_i| \leq 10^{4}∣A​i​​∣≤10​4​​ 。

思路：

O（n^2）暴力求区间和，然后统计答案到ans【i】中。

那么更新ans【i】=max（ans【i+1】，ans【i】）；

Ac代码：

#include
#include
#include
using namespace std;
int sum[150000];
int a[150000];
int ans[15000];
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)ans[i]=-0x3f3f3f3f;
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        for(int i=1;i<=n;i++)
        {
            if(i==1)sum[i]=a[i];
            else
            sum[i]=sum[i-1]+a[i];
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=i;j<=n;j++)
            {
                ans[j-i+1]=max(ans[j-i+1],sum[j]-sum[i-1]);
            }
        }
        for(int i=n-1;i>=1;i--)
        {
            ans[i]=max(ans[i],ans[i+1]);
        }
        while(m--)
        {
            int q;
            scanf("%d",&q);
            printf("%d\n",ans[q]);
        }
    }
}