ARC112A - Many Formulae

A - Many Formulae

题目描述

给定 $n$ 个数，在相邻两数之间加上+号或-号构成算式，不能出现连续两个减号。

求所有情况下式子结果的总和。

数据范围与提示

$1\le n\le 10^5，1\le a_i\le 10^9$ 。

前言

这两天昏昏沉沉的，不知道自己在想些什么。

A题做了90分钟，整场正式比赛直接垮掉。

令我感到更悲哀的是，我每次打ARC的正式比赛，总会在第一题卡住至少一小时（上一次是70多分钟时过掉第一题），后面的题反而想得挺快的。

不太可能是能力的问题，我也不相信我的能力这么畸形，所以只能是心态出了问题。

也许是很久没碰键盘，看到题就紧张了吧。

思路

全场最难的题

我看到这题立马想到了 一道类似的题 ，但是又没有回忆完，只记得用组合数，然后就推出了一个感觉正确的做法，怎么也过不了第三个样例（能过第一二个样例就离谱！），然后改了很久，一次比一次接近了，但是总是不相等。最后改到一个小时的时候实在改不下去了，看了看后面的题，顺手过了B题，然后C题有了点着落，写了个代码交上去 T 了（后来考试刚结束我就过了C题，那是后话），然后又回来看A题。

然后我终于发现我的做法假了，那道题相当于是限定了减号个数的特殊情况，特殊情况特殊做法。如果把那道题的做法强行移动到这道题上，不仅极其复杂，而且是 $O(n^2)$ 的过不了。

心态已经崩了。旁边的 $\mathrm{O}\mathrm{n}\mathrm{e}\mathrm{I}\mathrm{n}\mathrm{D}\mathrm{a}\mathrm{r}\mathrm{k}$ 已经过了五道题，我还只有一道题。他在我头撞电脑的声音中丝毫没被打乱。

杂乱的思路终于指使我跳出原题的情景，往递推上想，然后我终于发现这题其实是更简单的一种情况，DP式子很简单，而且就是斐波那契数列！

---

具体来说，设 $dp[i]$ 表示 $i$ 个位置上放任意个数减号不相邻的方案数，然后显然有 $dp[i]=dp[i-1]+dp[i-2]$ 。

做法肯定是先枚举每个数有多少种情况会被减去，显然情况数 $f(i)=dp[\max (0,i-3)]\ast dp[\max (0,n-i-1)]$ ，

然后用 $dp[n-1]\ast (\sum a_i)-\sum a_i\ast 2\ast f(i)$ 求得答案，复杂度 $O(n)$ 。

其它做法可能有，但是应该没有更优的做法了，大部分AC代码也都是这个做法。难道可以优化到比 $O(n)$ 更快吗？

---

我不知道是怎么把一道这么简单的题（ARC的A题）和一道很难的题（另一场ARC的F题）联系起来的。

如果是我想难了，又怎么会想到做法，不验证，就直接开打？

如果是我想简单了，又怎么会一直认为它与那道难题类似？

代码

上面有我求组合数的残留。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<ctime>
#define ll long long
#define MAXN 100005
#define uns unsigned
#define MOD 1000000007ll
#define INF 0x3f3f3f3f
using namespace std;
inline ll read(){
	ll x=0;bool f=1;char s=getchar();
	while((s<'0'||s>'9')&&s>0){if(s=='-')f^=1;s=getchar();}
	while(s>='0'&&s<='9')x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return f?x:-x;
}
inline ll ksm(ll a,ll b,ll mo){
	ll res=1;
	for(;b;b>>=1,a=a*a%mo)if(b&1)res=res*a%mo;
	return res;
}
int n;
ll a[MAXN],sum;
ll fac[MAXN<<1],inv[MAXN<<1];
inline ll C(int n,int m){
	if(m>n||m<0)return 0;
	return fac[n]*inv[m]%MOD*inv[n-m]%MOD;
}
ll dp[MAXN];
signed main()
{
	n=read();
	for(int i=1;i<=n;i++)a[i]=read(),sum=(sum+a[i])%MOD;
	
	fac[0]=fac[1]=inv[0]=inv[1]=1;
	for(int i=2;i<=(n<<1);i++)fac[i]=fac[i-1]*i%MOD;
	inv[n<<1]=ksm(fac[n<<1],MOD-2,MOD);
	for(int i=(n<<1)-1;i>1;i--)inv[i]=inv[i+1]*(1ll+i)%MOD;
	
	if(n==1){printf("%lld\n",a[1]%MOD);return 0;}
	if(n==2){printf("%lld\n",a[1]*2%MOD);return 0;}
//	printf(" %lld\n",ans);
	dp[0]=1,dp[1]=2;
	for(int i=2;i<=n;i++)dp[i]=(dp[i-1]+dp[i-2])%MOD;
	ll ans=dp[n-1]*sum%MOD;
	for(int i=2;i<=n;i++){
		ll ch=dp[max(0,i-3)]*dp[max(0,n-i-1)]%MOD;
		ans=(ans-ch*a[i]%MOD*2%MOD+MOD)%MOD;
	}
	printf("%lld\n",(ans+MOD)%MOD);
	return 0;
}