2021“MINIEYE杯”中国大学生算法设计超级联赛（5）

1009 Array（树状数组）

描述：
给定序列，找存在绝对众数的区间 (区间的众数出现次数大于区间长度一半) 个数。

分析：
本题与洛谷上一题除了数据范围是一模一样的。
其题解。
这题同样用线段树求理论上是可以过的，但是线段树常数可能大一下，以至于会被卡TLE 但是实验室有人线段树过了。
考虑用常数更小一点的树状数组。
要求是要维护-1，1序列的sum数组，做到区间修改，区间查询二阶前缀和。
然后在树状数组中要想做到区间修改，只能对差分数组进行单点修改，这样原序列就变成了差分数组的一阶前缀和，要实现的区间查询二阶前缀和变成了三阶前缀和。
设 a，b，c，d分别为原数组(差分数组)，一阶前缀和数组，二阶前缀和数组，三阶前缀和数组，那么有：

$d[x]={\sum }_{i=1}^x{\sum }_{j=1}^i{\sum }_{k=1}^{j}a[k]$

然后分析一下这个式子：

${\sum }_{j=1}^i{\sum }_{k=1}^{j}a[k]$

$=a_1+$
$a_1+a_2+$
$a_1+a_2+a_3+$
$...+$
$a_1+a_2+a_3+...+a_i$

$=i{a}_1+(i-1)a_2+...+a_i$

所以就有

$d[x]=$

$a_1+$
$2a_1+a_2+$
$...+$
$i{a}_1+(i-1)a_2+...+a_i$
$...+$
$x{a}_1+(x-1)a_2+...+2a_{x-1}+a_x$

可以发现最终$a_1,a_2,...,a_x$的系数式等差数列的前缀和，运用等差数列求和公式，有：

$d[x]={\sum }_{i=1}^x\frac{(x-i+2)(x-i+1)}{2}{a}_i$

$=\frac{(x+1)(x+2)}{2}{\sum }_{i=1}^x{a}_i-\frac{2x+3}{2}{\sum }_{i=1}^{x}i{a}_i+\frac{1}{2}{\sum }_{i=1}^x{i}^2{a}_i$

最后像线段树做法一样加上一个偏移量，做区间修改区间查询就可以了。

代码

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pli pair<ll,int>
#define Min(a,b,c)  min(a,min(b,c))
#define Max(a,b,c)  max(a,max(b,c))

typedef long long ll;
typedef unsigned long long ull;

const double pi = 3.141592653589793;
const double eps = 1e-8;
const int INF = 0x3f3f3f3f;

const int N = 1000010;

int n, del, a[N];
vector<int> vec[N];
ll c1[N << 1], c2[N << 1], c3[N << 1];

ll ask(int x)
{
    ll ans = 0;
    for (int i = x; i; i -= i & -i)
        ans += (ll)(x + 1) * (x + 2) * c1[i] - (ll)(2 * x + 3) * c2[i] + c3[i];
    return ans / 2;
}

void change(int x, int d)
{
    for (int i = x; i <= n + del; i += i & -i)
    {
        c1[i] += d;
        c2[i] += (ll)x * d;
        c3[i] += (ll)x * x * d;
    }
}

int main()
{   
    
    int t;
    scanf("%d", &t);    
    while (t--)
    {
        scanf("%d", &n);
        int mx = 0, mn = INF;
        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
            mx = max(mx, a[i]);
            mn = min(mn, a[i]);
            vec[a[i]].push_back(i);
        }
    
        del = n + 2;
        ll ans = 0;
    
        for (int i = mn; i <= mx; i++)
        {
            if (vec[i].empty()) continue;
            vec[i].push_back(n + 1);

            int pre = 0, sum = 0;
            for (int j = 0; j < vec[i].size(); j++)
            {
                int now = vec[i][j] - 1;
                int low = sum - (now - pre), up = sum;
                ans += ask(up - 1 + del) - ask(low - 2 + del);
                change(low + del, 1);
                change(up + 1 + del, -1);
                pre = now + 1;
                sum = low + 1;
            }
            
            pre = 0, sum = 0;
            for (int j = 0; j < vec[i].size(); j++)
            {
                int now = vec[i][j] - 1;
                int low = sum - (now - pre), up = sum;
                change(low + del, -1);
                change(up + 1 + del, 1);
                pre = now + 1;
                sum = low + 1;
            }
           vec[i].clear();   
        }

        printf("%lld\n", ans);
    }
    return 0;
}