浅谈一类组合问题的差分解法

先来一道题目：2020ICPC模拟赛A题

题意很简单，给定$a,b,c,d$，求整数$x,y,z,k$满足$0\le x\le a,0\le y\le b,0\le z\le c,0\le k\le d$且$x+y+z=k$的解的数量。

首先可以考虑更简单的题意，即求$x+y=z$的解的数量，

这个也不会，怎么办啊

那就考虑最简单的问题：已知整数$0\le x\le a,0\le y\le b$，如何求$x=y$的方案数，

这个就是有手就行了，很显然，答案是$x,y$取值范围的交，

更具体地说，设$w[i]$为$x$在$i$处的取值方案数，则总的方案数就是${\sum }_{i=0}^{b}w[i]$，

由于$x$在$[0,a]$内的每个整数点都有一种取法，即$w[0]=w[1]=...=w[a]=1$，我们就可以直接用一次差分更新$w[0:a]$了，从而两次前缀和就求出了答案。

进一步考虑$x+y=z$的情况（其中$0\le x\le a,0\le y\le b,0\le z\le c$），

设$w[i]$为$x+y=i$的方案数，则总的方案数就是${\sum }_{i=0}^{c}w[i]$，

和上一个问题不同，对于取值范围内不同的$i$，$w[i]$的取值可能不相同，也就是说一次差分是届不到的，

因此我们考虑枚举$x$，对于每个不同的$x$求出此时所有可能的$y$对$w[x+y]$的贡献，

设$v[i]$为$y$在$i$处取值的方案数，显然$v[0]=v[1]=...=v[b]=1$，

对于一个$x$，$y$的所有取值分别会给$w[x],w[x+1],...,w[x+b]$增加$v[0],v[1],...,v[b]$的贡献，

又因$v[0]=v[1]=...=v[b]=1$，我们又可以用差分进行区间加操作了，

每次枚举一个$x$，差分更新$w[x]$到$w[x+b]$的值，枚举完所有的$x$后就通过两次前缀和求出了答案。

接下来回到$x+y+z=k$的情况，已经可以通过这种方法解决了，

考虑先枚举$x$，再枚举$x+y$和$x+y+z$，每次枚举结束后通过前缀和计算贡献和答案即可。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3000010;
ll a,b,c,d,ans;
ll v[N],w[N];//v[i]为x+y=i的方案数,w[i]为x+y+z=i的方案数
int main(){
    ios::sync_with_stdio(false);
    cin>>a>>b>>c>>d;
    for(int i=0;i<=a;i++){
        v[i]++;//对每个x更新[x,x+b]区间
        v[i+b+1]--;
    }
    for(int i=1;i<=a+b;i++){
        v[i]+=v[i-1];//前缀和求v[i]
    }
    for(int i=0;i<=a+b;i++){
        w[i]+=v[i];//对每个x+y更新[x+y,x+y+c]区间
        w[i+c+1]-=v[i];
    }
    for(int i=1;i<=a+b+c;i++){
        w[i]+=w[i-1];//前缀和求w[i]
    }
    for(int i=0;i<=d;i++){
        ans+=w[i];//对应k=i的情况加到答案上
    }
    cout<<ans;
}

杂谈：

其实这个东西是Awei几个月前做某道CF题在别人代码里看到的方法，当时没怎么看懂，昨晚CF赛后，她他又跟我说起了这个（E题能用到），睡前终于想明白了，于是就写了篇博客（