CF1467D Sum of Paths dp

题目链接

题意：
给你一个长度为$n$的序列，其中第$i$个位置权值为$a_i$，要有$k$次移动，每次移动向左或向右移动一格，不能出边界（到$0$或者$n+1$），对于这样一个连续移动$k$次后得到的长度为$k+1$的经过位置序列，我们称之为好序列，它的权值是经过的每一个位置的$a_i$之和，其中多次经过的话要多次累加。现在对于所有起始位置，求所有好序列的权值和，并且有$q$次对序列权值的修改。答案对$1e9+7$取模。
$n,k<=5000,q<=200000,a_i<=1e9$

题解：
看到带修改，一个比较容易得到的想法是考虑算出每一个位置在所有情况中一共被经过了多少次，然后修改只需要改一下乘的$a_i$即可。

那么我们来考虑怎么求每个位置在所有好序列中出现的总次数。这里计数的对象选取就十分关键了。这个题的打开方式是，对于每一个位置$i(1<=i<=n)$，我们枚举它在好序列中是第几步经过的，那么一个位置被经过的总次数是${\sum }_{j=0}^k第j步经过位置i的方案数$。

那么问题又变成了如何对于每一个$i$，求出第$j$步经过它的方案数。我们设$dp[i][j]$表示走了$j$步，到达点$i$的所有方案数。我们可以将每条路径都反向，可以建立双射，由此可知$dp[i][j]$也是由$i$出发，走了$j$步的合法序列的方案数。那么我们要求的第$j$步经过位置$i$的方案数就是$dp[i][j]\ast dp[i][k-j]$。具体含义是相当于走$j$步到$i$的方案数乘从$i$出发走完剩下$k-j$步的方案数。

最后我们只需要知道求出所有$dp[i][j]$。显然一开始对于每一个$i$，都有$dp[i][0]=1$，表示走$0$步到第$i$个位置的方案数为$1$。转移是$dp[i][j]=dp[i-1][j-1]+dp[i+1][j-1]$，注意$i=1$或$i=n$时要特判，不能跑到边界外。

这样这个题就做完了，累加答案之后对于每一次修改，根据出现次数和前后权值修改答案即可。

由于放寒假失踪人口回归了，所以有代码了。

代码：

#include <bits/stdc++.h>
using namespace std;

long long mod=1e9+7;
int n,k,q;
long long a[10010],dp[5010][5010],cnt[10010],ans;
int main()
{
	scanf("%d%d%d",&n,&k,&q);
	for(int i=1;i<=n;++i)
	scanf("%d",&a[i]);
	for(int i=1;i<=n;++i)
	dp[i][0]=1;
	for(int j=1;j<=k;++j)
	{
		for(int i=1;i<=n;++i)
		{
			if(i==1)
			dp[i][j]=dp[i+1][j-1];
			else if(i==n)
			dp[i][j]=dp[i-1][j-1];
			else
			dp[i][j]=(dp[i-1][j-1]+dp[i+1][j-1])%mod;
		}
	}
	for(int i=1;i<=n;++i)
	{
		for(int j=0;j<=k;++j)
		cnt[i]=(cnt[i]+dp[i][j]*dp[i][k-j]%mod)%mod;
	}
	for(int i=1;i<=n;++i)
	ans=(ans+cnt[i]*a[i]%mod)%mod;
	for(int i=1;i<=q;++i)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		ans=(ans-a[x]*cnt[x]%mod+mod+y*cnt[x]%mod)%mod;
		a[x]=y;
		printf("%d\n",ans);
	}
	return 0;
}