[Atcoder Grand 006 C] Rabbit Exercise 解题报告 (期望)

题目链接：https://www.luogu.org/problemnew/show/AT2164 https://agc006.contest.atcoder.jp/tasks/agc006_c

题目

中文描述：

?只兔子排成了一行, 第?只兔子的初始位置为??.

兔子要进行锻炼, 每一轮包含?次跳跃, 第?次是兔子 ??(2≤??≤?−1)进行跳跃, 它会等概率的选择第??−1 或第??+1只兔子 (假设为?), 然后跳到当前位置关于?的 对称点上.

这?次跳跃重复进行了?轮, 求最终每只兔子坐标的期望.

解题方法：

第?只兔子跳过第?只兔子之后坐标变为2∗?_?−?_?, 是线性的.

所以设??为第?只兔子位置的期望, 这个式子也成立.

进行完一次跳跃后,??变成了1/2$(2x_{i-1} - x_{i-1})$ + 1/2(2$x_{i+1}$ − $x_i$) = $x_{i-1} + x_{i+1}$ - $x_i$

设s?=?_?+1−?_?, 那么进行一次跳跃相当于交换s_?和 s_?-1.

?_?−1 x_i ?_?+1 -> ?_?−1 ?_?−1+x_i+1-x_i x_i+1

s数组: x_i-x_i-1 x_i+1-x_i-> x_i+1-x_i x_i-x_i-1

我们先手动做一轮，得到交换后对应的位置。考虑dfs一遍找到形成的交换环，对每只兔子在它的环上走k步，最后再把差分的值加上就好.

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;

const int maxn=1e5+15;
ll n,m,k,cnt;
ll vis[maxn],s[maxn],x[maxn],num[maxn],be[maxn],p[maxn],a[maxn];
vector <ll> q[maxn];
void dfs(ll x,ll dep)//找环 
{
    num[x]=dep;vis[x]=1;be[x]=cnt;
    q[cnt].push_back(x);
    if (!vis[p[x]]) dfs(p[x],dep+1);
}
int main()
{
    scanf("%lld",&n);
    for (int i=1;i<=n;i++) scanf("%lld",x+i);
    for (int i=1;i<n;i++) s[i]=x[i+1]-x[i];//差分序列 
    scanf("%lld%lld",&m,&k);
    for (int i=1;i<n;i++) p[i]=i;
    for (int i=1;i<=m;i++) {scanf("%lld",a+i);swap(p[a[i]],p[a[i]-1]);}//一次置换的结果，记录下一次置换兔子到哪了 
    for (int i=1;i<n;i++) if (!vis[i]) {dfs(i,0);++cnt;}
    ll now=x[1];
    printf("%lld\n",now);
    for (int i=1;i<n;i++)
    {
        now+=s[q[be[i]][(num[i]+k)%q[be[i]].size()]];
        printf("%lld\n",now);
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/xxzh/p/9343418.html