【gdgzezoi】Problem B: Rabbit Exercise

Description

有N只兔子站在数轴上。为了方便，将这些兔子标号为1…N。第i只兔子的初始位置为xi。

现在这些兔子会按照下面的规则做若干套体操。每一套体操由M次跳跃组成；在第j次跳跃的时候，第aj(2≤aj≤N−1)只兔子会等概率随机选择第aj−1或aj+1只兔子中的一只（不妨设选择了第x只兔子），然后跳当前位置到关于第x只兔子对称的点。

这些兔子会按顺序做K套相同的体操。现在请你求出，每一只兔子做完K套体操之后最终位置坐标的期望值。
Input
题目数据按以下格式从标准输入输出输入：

N
x1 x2 … xN
M K
a1 a2 … aM
Output
总共输出N行。第i行为第i只兔子做完K套体操之后的位置坐标的期望。输出保留一位小数(四舍五入)。
Sample Input
sample input 1:
3
-1 0 2
1 1
2

sample input 2:
3
1 -1 1
2 2
2 2

sample input 3:
5
0 1 3 6 10
3 10
2 3 4
Sample Output
sample output 1:
-1.0
1.0
2.0

sample output 2:
1.0
-1.0
1.0

sample output 3:
0.0
3.0
7.0
8.0
10.0
HINT

• 3≤N≤105
• xi为整数。
• |xi|≤109
• 1≤M≤105
• 2≤aj≤N−1
• 1≤K≤1018

样例 1 ：

第2只兔子会进行一次跳跃。如果选择第1只兔子，跳跃终点的坐标为−2；如果选择第3只兔子，跳跃终点的坐标为4。所以，第2只兔子最终坐标的期望是0.5×(−2)+0.5×4=1.0

样例 2：

xi可能会相同。

思路

差分，在第x位做一个这样的操作，等价于把x和x+1位的值互换。这样我们直接把M次操作看成是M个对换，做完这些操作就可以看成是1到n的一个置换。我们我们直接把置换拆成一些轮换，然后直接在每个轮换上面每个点走K步就行啦。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a[100010],ans[100010];
int c[100010],b[100010],d[100010]; 
int main()
{
    int n;
    scanf("%d",&n);
    int i;
    ll sum=0;
    for(i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        a[i]-=sum;
        sum+=a[i];
        c[i]=i;
    }
    int m;
    ll k;
    scanf("%d%lld",&m,&k);
    int x;
    for(i=1;i<=m;i++)
    {
        scanf("%d",&x);
        swap(c[x],c[x+1]);
    }
    for(i=1;i<=n;i++)
        if(!b[i])
        {
            int cnt=0;
            int j;
            for(j=i;!b[j];j=c[j])
            {
                b[j]=1;
                d[++cnt]=j;
            }
            for(j=1;j<=cnt;j++)
                ans[d[j]]=a[d[(j+k-1)%cnt+1]];
        }
    for(i=1;i<=n;i++)
    {
        ans[i]+=ans[i-1];
        printf("%lld.0\n",ans[i]);
    }
    return 0;
}