Ahoi2005 洗牌 扩展欧几里得

Ahoi2005 洗牌

问题描述

为了表彰小联为Samuel星球的探险所做出的贡献，小联被邀请参加Samuel星球近距离载人探险活动。 由于Samuel星球相当遥远，科学家们要在飞船中度过相当长的一段时间，小联提议用扑克牌打发长途旅行中的无聊时间。玩了几局之后，大家觉得单纯玩扑克牌对于像他们这样的高智商人才来说太简单了。有人提出了扑克牌的一种新的玩法。 对于扑克牌的一次洗牌是这样定义的，将一叠N（N为偶数）张扑克牌平均分成上下两叠，取下面一叠的第一张作为新的一叠的第一张，然后取上面一叠的第一张作为新的一叠的第二张，再取下面一叠的第二张作为新的一叠的第三张……如此交替直到所有的牌取完。 如果对一叠6张的扑克牌1 2 3 4 5 6，进行一次洗牌的过程如下图所示：

从图中可以看出经过一次洗牌，序列1 2 3 4 5 6变为4 1 5 2 6 3。当然，再对得到的序列进行一次洗牌，又会变为2 4 6 1 3 5。 游戏是这样的，如果给定长度为N的一叠扑克牌，并且牌面大小从1开始连续增加到N（不考虑花色），对这样的一叠扑克牌，进行M次洗牌。最先说出经过洗牌后的扑克牌序列中第L张扑克牌的牌面大小是多少的科学家得胜。小联想赢取游戏的胜利，你能帮助他吗？

输入格式

有三个用空格间隔的整数，分别表示N，M，L （其中0＜ N ≤ 10 ^ 10 ，0 ≤ M ≤ 10^ 10，且N为偶数）。

输出格式

单行输出指定的扑克牌的牌面大小。

样例输入

6 2 3

样例输出

6

---

拿到题目有两种想法：

第一种：草稿纸或打表推出N比较小的时候多次洗牌后的情况，找规律。

这种想法是比较常规的。观察可得洗牌是会出现周期的，虽然周期的长度没有规律，但是一般都比较小。那么我们可以先暴力跑一次周期，然后就可以缩小M的范围并且得到答案。这样的做法可以得到很高的分数。

但是，“一般都比较小”总让我们心里没底，万一周期非常长呢？事实上确实会有很长的周期，所以这样的做法不是标准解法。但是能写出这样的解法还是不错的。

第二种：直接推出一般情况下的数学解法

既然本题答案求的是某个位置上的数，那么自然会想到位置的变化。根据洗牌的原理，很容易发现，对前$\frac{N}{2}$段和后$\frac{N}{2}$段的位置变换规则是不一样的。具体来讲，很容易推出规则是这样的：

设当前状态下的位置为$X(1\leq X \leq N)$，洗一次牌后的位置为$X'$。

当$X\leq \frac{N}{2}$时，$X'=2X$。
当$X> \frac{N}{2}$时，$X'=2(X-\frac{N}{2})-1=2X-N-1$。

然而接下来就变得不好处理了：每次还要对$X$的大小进行分类讨论。如果到这里就放弃这种做法，那就放弃了正解。

如果不用分类讨论，那么似乎会变得比较简单。是否可以去掉分类讨论的步骤呢？其实，考虑到每种情况下$X$的取值范围，两种情况可以合并成一种：

$$X'=2Xmod(N+1)$$

剩下的就非常好处理了。由于开始的状态下，牌面数字等于位置，所以我们只需要用扩展欧几里得算法求出下面方程的$X$即可：

$$2^MX\equiv L(mod (N+1))$$

---

代码：

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

ll N,M,L,Ans;

ll ksm(ll a,ll b,ll c)
{
    ll ans=1;a%=c;
    while(b)
    {
        if(b&1)ans=ans*a%c;
        b>>=1;a=a*a%c;
    }
    return ans;
}

ll exgcd(ll a,ll b,ll &x,ll &y)
{
    if(!b){x=1;y=0;return a;}
    ll r=exgcd(b,a%b,x,y),tmp;
    tmp=x;x=y;y=tmp-a/b*y;
    return r;
}

int main()
{
    ll x,y,d;

    cin>>N>>M>>L;
    d=exgcd(ksm(2,M,N+1),N+1,x,y);
    Ans=x*L/d%(N+1);
    Ans=(Ans+(N+1))%(N+1);
    cout<<Ans;
}