智乃酱的静态数组维护问题多项式

本文讲解了如何利用差分维护技术,针对最高次数为5的多项式进行6次差分来快速修改指定区间,并通过6次前缀和转换回原数组,实现对多项式函数区间改动的O(1)时间复杂度。关键步骤包括理解差分与前缀和的关系,以及如何抵消额外影响的计算策略。

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

差分维护多项式前缀和

在这里插入图片描述

前置知识

定理:最高次项为n次的多项式做n+1次差分的余项为常数

题目最多为5次,所以做6次差分就够了。 如果次数小于6次,每多做一次差分增加一项余项,余项还是不超过6(可以直接去打印出来试试看)



现在开始解题

题目要求修改区间 l ∼ r l \sim r lr
实际上是给这个区间各个位置加上对应的多项式函数 0 , ⋯ , f ( x 1 ) , f ( x 2 ) , ⋯   , f ( x i ) , ⋯   , f ( x r − l + 1 ) 0 , \cdots, f(x_{1}) , f(x_{ 2}) , \cdots , f(x_{i}) , \cdots, f(x_{r - l + 1}) 0f(x1),f(x2),,f(xi),,f(xrl+1)
这样的时间复杂度是o(n)的,这个时候可以利用上面的那个定理,对原数组和多项式分别做6次差分再把他转换成修改区间
l ∼ l + 5 l\sim l + 5 ll+5
因为差分和前缀和是两个逆过程,原数组求差分可以得到差分数组,差分数组求前缀和可以得到原数组,差分数组和原数组可以相互转换。 所以可以把对原数组的区间修改转换为对差分数组的修改,修改结束后再把差分数组做对应次数的前缀和转换为原数组即可

题目的多项式最高次数只有5,对多项式的 1 ∼ r − l + 1 1\sim r - l + 1 1rl+1做6次差分,就能保证余项一定在前6项,后面的项就都是0,对结果不影响,所以你只要把多项式6次差分得到的前6项修改到原数组的6次差分数组的前6项即可,这样时间复杂度就是O(1)




细说一下:你把区间 l ∼ r l \sim r lr 各自加上一个多项式 f ( i ) , i ϵ [ 1 , r − l + 1 ] f(i),i \epsilon \left [ 1,r - l + 1 \right ] f(i),iϵ[1,rl+1]
可以转换为先对多项式求6次差分,然后转换为对差分数组(做了6次差分后的原数组)的 l ∼ l + 5 l \sim l + 5 ll+5 进行修改。
然后若干次修改后,再做6次前缀和把差分数组还原成原数组即可。



重点来了

因为差分,所以要考虑对 r + 1 ∼ n r + 1 \sim n r+1n 的影响,所以要做一个抵消操作。

如何抵消?

假设现在对 l ∼ r l \sim r lr的区间,我们只做了对差分数组的
l ∼ l + 5 l \sim l + 5 ll+5的修改操作(假设原数组全0),这样直接做6次前缀和得到的结果为
0 , ⋯ , f ( x 1 ) , f ( x 2 ) , ⋯   , f ( x i ) , ⋯   , f ( x r − l + 1 ) , f ( x r − l + 2 ) , ⋯   , f ( x r − l + j ) , ⋯   , f ( x n − l + 1 ) 0 , \cdots, f(x_{1}) , f(x_{ 2}) , \cdots , f(x_{i}) , \cdots, f(x_{r - l + 1}) ,f( x_{r - l + 2}) , \cdots , f(x_{r -l + j}) , \cdots , f(x_{n - l + 1}) 0f(x1),f(x2),,f(xi),,f(xrl+1),f(xrl+2),,f(xrl+j),,f(xnl+1)

多加了
f ( x r − l + 2 ) , ⋯   , f ( x r − l + j ) , ⋯   , f ( x n − l + 1 ) f( x_{r - l + 2}) , \cdots , f(x_{r -l + j}) , \cdots , f(x_{n - l + 1}) f(xrl+2),,f(xrl+j),,f(xnl+1)

如何抵消?我们给
r + 1 ∼ n r + 1\sim n r+1n区间也做一次修改减去这些对应的项就行,也就是相当于对
r + 1 ∼ n r + 1 \sim n r+1n减去 f ( x r − l + 2 ) , ⋯   , f ( x r − l + j ) , ⋯   , f ( x n − l + 1 ) f( x_{r - l + 2}) , \cdots , f(x_{r -l + j}) , \cdots , f(x_{n - l + 1}) f(xrl+2),,f(xrl+j),,f(xnl+1)

这个时间复杂度也是O(n)的,如何降低?很简单用同样方法就行!

你可以把这个修改也转换成做了6次差分后的修改。变成对做了6次差分后的原数组的
r + 1 ∼ r + 6 r + 1 \sim r + 6 r+1r+6 减去做过6次差分后

f ( x r − l + 2 ) , ⋯   , f ( x r − l + j ) , ⋯   , f ( x n − l + 1 ) f( x_{r - l + 2}) , \cdots , f(x_{r -l + j}) , \cdots , f(x_{n - l + 1}) f(xrl+2),,f(xrl+j),,f(xnl+1)
即可。

因为这个是直接从r + 1做到n(末尾),已经没有剩余的区间受到影响了,所以直接做就行。

代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 20;
typedef long long ll;
const ll mod = 1e9 + 7;
ll a[N];
ll ceo1[12];
ll ceo2[12];
ll xishu[10];

ll M(ll x,ll y){//取模
    return ( (x + y) % mod + mod ) %mod;
}

void D(ll d[],int len, int ct){// 对长度为 len 的 d 数组求 ct 次差分
    while(ct--){
        for(int i = len; i ; --i)
            d[i] = M(d[i], - d[i-1]);
            
    }
}
void F(ll d[],int len, int ct){//对长度为 len 的 d 数组求 ct 次前缀和
    while(ct--){
        for(int i = 2; i <= len; ++i)
            d[i] = M(d[i], d[i-1]);
    }
}

ll G(ll x,ll k){//求最高次项为 k 的 f(x)
    ll res = 0,base = 1;
    for(int i = 0; i <= k; ++i)
    {
        res = M(res , base * xishu[i] % mod);
        base = base * x % mod;
    }
    return res;
}

void get(ll d[], int l, int k, ll f){// 求得要修改的l ~ r的 前6项(l ,l + 1, …… l + 5)
    for(int i = 1; i <= 6; ++i){
        d[i] = f * G(l + i - 1,k);
    }
}

int main(){
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n,m,q;  cin >> n >> m >> q;
    for(int i = 1; i <= n; ++i)
        cin >> a[i];
    D(a,n,6);
    while(m--) {
        int l,r,k;  cin >> l >> r >> k;
        for(int i = 0; i <= k; ++i)//输入系数
            cin >> xishu[k - i];
        get(ceo1, 1, k, 1);//获取 l ~ l + 5 对原数组的修改
        get(ceo2, r - l + 2, k, -1);//获取 r + 1 ~ n 对原数组的修改
        D(ceo1,6,6);// 将 l ~ l + 5 对原数组的修改求6次差分
        D(ceo2,6,6);// 将 r + 1 ~ n 对原数组的修改求6次差分
        for(int i = 1; i <= 6; ++i)//更新对差分数组的修改
        {
            a[i + l - 1] = M(a[i + l - 1],ceo1[i]);
            a[r + i] = M(a[r + i], ceo2[i]);
        }
        
    }
    F(a, n, 7);//做 6 + 1次前缀和,因为要求 l ~ r的区间和,所以多做一次前缀和
    
    while(q--) {
        int l,r;    cin >> l >> r;
        cout << M(a[r], -a[l - 1] ) << endl;
    }
    
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值