Intercity Travelling CodeForces

Intercity Travelling CodeForces"/>

Intercity Travelling CodeForces

大意: 有一段$n$千米的路, 每一次走$1$千米, 每走完一次可以休息一次, 每连续走$x$次, 消耗$a[1]+...+a[x]$的能量. 休息随机, 求消耗能量的期望$\times 2^{n-1}$.

简单计数题, 枚举每种长度的贡献.

#include <iostream>
#include <algorithm>
#include <cstdio>
#define REP(i,a,n) for(int i=a;i<=n;++i)
using namespace std;
typedef long long ll;
const int P = 998244353, INF = 0x3f3f3f3f;const int N = 1e6+10;
int n,a[N],po2[N];int main() {scanf("%d", &n);REP(i,1,n) scanf("%d", a+i),(a[i]+=a[i-1])%=P;if (n==1) return printf("%d\n",a[n]),0;if (n==2) return printf("%d\n",2*a[1]+a[2]),0;po2[0] = 1;REP(i,1,n) po2[i] = po2[i-1]*2ll%P;int ans = a[n];REP(i,1,n-1) ans = (ans+((n-i-1ll)*po2[n-i-2]%P+2ll*po2[n-i-1])*a[i])%P;printf("%d\n", ans);
}

转载于:.html