AcWing 5386. 进水出水问题【线性dp+差值dp】

原题链接：https://www.acwing.com/problem/content/5389/

某泳池装有 n 个水管，编号 1∼n。

每个水管都是既可用于进水，也可用于出水。

其中，第 i 个水管工作时的单位时间进水或出水量为 ai。

我们希望泳池保持水循环的同时，还能够保持水位不变。

为此，我们需要制定一种水管工作方案。

具体要求为：

1. 选择若干个（至少一个）连续编号的水管作为工作水管（未选择的水管不工作）。
2. 每个工作水管，要么用于进水，要么用于出水。
3. 所有安排完毕以后，所有用于进水的工作水管的进水量之和应当等于所有用于出水的工作水管的出水量之和。

请你计算，一共可以制定出多少种不同的水管工作方案。

注意，（1 号水管进水、2 号水管出水）和（1 号水管出水、2 号水管进水）应当视为两种不同的方案，虽然它们用到的工作水管相同，但是水管的具体安排（进水或出水）不同。

输入格式

第一行包含整数 n。

第二行包含 n 个整数 a1,a2,…,an。

输出格式

一个整数，表示可以制定出的不同方案的总数量对 1e9+7 取模后的结果。

数据范围

前 3 个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤1000，1≤ai≤1000，ai<=10000(1<=i<=n)

输入样例：

4
1 1 1 1

输出样例：

12

解题思路：

首先看到这种求方案数，又有方案数可能很大需要取模，那么这个题目十有八九就是可以推出数学公式或者就是dp,，但是我没有发现可以推出什么公式，但是从题目可以看到俩个操作，对于选定工作的水管，要么用于进水，要么用于出水，这很像线性dp的常规操作，我首先想到的就是首先枚举所有区间然后对于每个区间进行线性dp，但是这个方法时间复杂度太高了，根本过不了， 而且也找不到很好的优化方式，看到n=1000,如果先枚举所有区间再dp显然是不现实的，我们可以看看有没有什么好的状态定义方式使得不需要暴力枚举所有区间，而是可以直接设置状态进行dp呢，我们可以发现题目要求选择的一段连续工作水管的进水量和出水量需要相同，涉及到俩个变量，进水量之和最多是0-10000,出水量之和最多是0-10000，同时要使得进水量之和和出水量之和相同就是要使得进水量和出水量的差值为0，这个时候我们看到了差值这个词，那么我们可以尝试使用差值设计状态，以前貌似没见过这种差值状态设计，所以这次确实没有想出来这样设计状态。

状态定义：

定义：f[i][j]表示以i结尾并且进水量和出水量的差值为j的所有方案

初始状态：

正常情况下，我们会设f[0][0]=1，但是这个题目不能这样设初始状态，如果这样设初始状态，当i==1时，会导致(1)(3)或者(2)(4)出现重复计算情况，所以我们不这样设初始状态，出现这种情况是因为当i==1时，第(3)种情况虽然定义的是前面还有管子，但是在当前是第一根管子时，前面根本就没有管子了，导致此时第(3)变为了和(1)一样，相当于这种情况被计算了俩次，导致了重复计算。我们将所有的f[i][a[i]]和f[i][-a[i]]初始值定为1即可，因为工作的水管必须是连续的，所以状态定义状态转移时不仅需要考虑当前位置进水或者出水，而且还需要考虑当前位置前面还有没有工作的水管，所以这就是为什么我们的初始状态不能按照常规方式去设的原因，也是常规做法状态转移只有俩种情况，但是这里有四种状态转移情况的原因。

状态转移：

(1)当前位置管子设为进水，并且前面没有工作管子了

f[i][a[i]]++;

(2)当前位置管子设为出水，并且前面没有工作管子了

f[i][-a[i]]++;

(3)当前位置管子设为进水，并且前面还有工作管子

f[i][j]=(f[i][j]+f[i-1][j-a[i]])

(4)当前位置管子设为出水，并且前面还有工作管子

f[i][j]=(f[i][j]+f[i-1][j+a[i]]);

最终答案

答案是所有的f[i][0]的和

注意由于我们定义的是差值，所以我们这个状态转移方程会出现负数下标，我们需要对状态转移方程做一个等价偏移，a数组所有数的和最大是10000，所以出现的负数最小值为-10000，我们设置偏移值为10000即可。

时间复杂度：第一维枚举水管，时间为O(n),第二位枚举差值时间为O(20000)，最终时间为1000*20000，大概是2e7，这个时间复杂度是可以过的。

空间复杂度：2e7/1e6*8=160，大概是160M,题目给了256M,所以空间是足够的，如果我们将f定义为int类型，那么可以将空间降低到80M，这个题目给了256M，所以f数组定义为long long 类型也是可以的。

cpp代码如下

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
typedef long long LL;

const int N=1010,M=20010,mod=1e9+7,B=10000;  //B为偏移值

int n;
int a[N];
LL f[N][M];  //f[i][j]表示以i结尾并且进水值和出水值之差为j的所有方案
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int res=0;
    for(int i=1;i<=n;i++)
    {   //开始的范围是-10000~10000，偏移之后就是0~20000
        for(int j=0;j<=20000;j++)
        {
            //当前位置进水
            if(j-a[i]>=0)f[i][j]=(f[i][j]+f[i-1][j-a[i]])%mod;
            //当前位置出水
            if(j+a[i]<=20000)f[i][j]=(f[i][j]+f[i-1][j+a[i]])%mod;
        }
        //设置f[i][a[i]]和f[i][-a[i]]的初始值为1
        f[i][B+a[i]]++,f[i][B-a[i]]++;
        res=(res+f[i][B])%mod;  //对于所有的f[i][0]求和就是答案，偏移之后是f[i][B]
    }
    cout<<res<<endl;
    return 0;
}

总结：之前貌似没有遇到过这种状态设计方式，所以当时没想出来，这次学会了，吸取教训，下次还遇到类似的希望能很快就想出来状态设计吧。