[HDU 5086] Revenge of Segment Tree （前缀和）

最新推荐文章于 2017-07-26 14:23:41 发布

原创最新推荐文章于 2017-07-26 14:23:41 发布 · 351 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#acm #前缀和

本文介绍了一种解决HDU5086题目中所有连续区间和的有效算法。通过计算两次前缀和并利用特定公式，快速得出答案。代码实现使用C++完成。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接

HDU 5086

题意

给出一段序列，求所有连续区间和。

题解

求两次前缀和sum[], sum2[]，答案就是total[sum[i]*i - sum2[i-1]]。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long lint;
const lint mod = 1000000007;
#define maxn (447777)
int a[maxn];
lint sum[maxn], sum2[maxn];
int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int N;
        cin >> N;
        for(int i = 0; i <= N; i++)
            sum[i] = sum2[i] = d[i] = 0;
        for(int i = 1; i <= N; i++)
            scanf("%d", &a[i]);
        sum[1] = a[1];
        for(int i = 2; i <= N; i++)
        {
            sum[i] += a[i] + sum[i-1];
            sum[i] %= mod;
        }
        sum2[1] = sum[1];
        for(int i = 2; i <= N; i++)
        {
            sum2[i] += sum[i] + sum2[i-1];
            sum2[i] %= mod;
        }
        lint o = a[1];
        for(int i = 2; i <= N; i++)
        {
            o += ((sum[i] * i) % mod + mod - sum2[i-1]) % mod;
            o %= mod;
        }
        cout << o << endl;
    }
    return 0;
}