ARC107 B - Quadruple（差分）

live4m

于 2021-03-26 20:52:08 发布

阅读量215

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/weixin_44178736/article/details/115255031

这篇博客介绍了如何通过差分预处理技巧高效计算当a+b-c-d等于常数k时，所有可能组合的方案数。博主详细展示了如何构造递推关系f(i+k)=k+f(i)，并利用这个关系进行O(n)复杂度的枚举求解，最终给出了一段C++代码实现。

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

题意：

在这里插入图片描述

解法：

$a+b−c−d=k(a+b)−(c+d)=k(a+b)=k+(c+d)设f(i)为a+b=i的方案数,那么式子可以变为:f(i+k)=k+f(i)f(i)可以利用差分预处理枚举i,答案累加f(i)∗f(i+k).a+b-c-d=k\\ (a+b)-(c+d)=k\\ (a+b)=k+(c+d)\\ 设f(i)为a+b=i的方案数,那么式子可以变为:\\ f(i+k)=k+f(i)\\ f(i)可以利用差分预处理\\ 枚举i,答案累加f(i)*f(i+k).$

code：

#include<bits/stdc++.h>
#define int long long
//#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
const int mod=998244353;
int f[maxm];
int n,k;
void solve(){
    cin>>n>>k;
    //f[i]为a+b=i的方案数,差分可以O(n)预处理
    for(int i=1;i<=n;i++){//枚举a,b的取值为[1,n]
        f[i+1]++;
        f[i+n+1]--;
    }
    for(int i=1;i<maxm;i++){
        f[i]+=f[i-1];
    }
    //
    int ans=0;
    for(int i=1;i<maxm;i++){
        if(k+i>=0&&k+i<maxm){
            ans+=f[i]*f[k+i];
        }
    }
    cout<<ans<<endl;
}
signed main(){
    solve();
    return 0;
}