【洛谷P4706】取石子

Description

  
​   现在 Yopilla 和 yww 要开始玩游戏!
  
  ​ 他们在一条直线上标记了 \(n\) 个点,从左往右依次标号为 \(1, 2, ..., n\) 。然后在每个点上放置一些棋子,其中第 \(i\) 个点放置了 \(a_i\) 个棋子。接下来,从 Yopilla 开始操作,双方轮流操作,谁不能操作谁输。每次的操作是:当前操作方选定一个有棋子的点 \(x\) ,然后选择至少一个点 \(x\) 上的棋子,然后把这些棋子全都移动到点 \(x / prime\) 上,其中 \(prime\) 是一个质数,且 \(prime \mid x\)
  
  ​ Yopilla 最初一次操作的策略是随机的:随机找到一个有棋子的点 \(x\) ,随机选择正整数个棋子 \(y\) ,随机转移到一个能转移到的点 \(z\) 。所有棋子可以看作是一样的,换句话说:两种操作不同,当且仅当三元组 \((x, y, z)\) 不同。之后双方都按照最优策略来操作。
  
​   Yopilla 想要预测,他能够获胜的概率是多少,答案对 \(998244353\) 取模。
  
​  
  
  
  

Solution

  
​   我们发现,对于每一个数,如果以其幂指数之和为下标来将它们重新排列成一个数组,这个问题就变成了阶梯\(Nim\)问题。一次操作,相当于将一个数移动到其左边。不能移动者输。
  
​   事实上我们不需要实现这个重排操作。我们只需要知道每个数重排后是否在奇位置即可。
  
​   记输入数列为\(a\),我们统计出所有处于奇位置的数\(x\)\(a_x\)的异或和\(sum\)
  
​   我们要统计Yopilla一开始的随机操作一共有多少种可能、以及总共有多少种可能,使得操作后局面的先手必败。前者很好计算,就是\(\sum_x a_x*b_x\),其中\(b_x\)表示\(x\)这个数的不同质因子个数。
  
​   后者如何计算呢?对操作分类:(1)移动奇位置的数至偶位置、(2)移动偶位置的数至奇位置。
  
​   我们枚举所有奇位置的数。假设对该位置\(i\)操作后,总异或和\(sum\)等于0,即操作后先手必败,则\(a_i\)应该由\(a_i\)变成\(target=sum\; \text{xor}\; a_i\)
  
  如果原值比目标值大,那么显然(1)容易满足,选出\(a_i-target\)个数,并将它们通过任意一个质因子移动到偶位置,一共有\(b_i\)种合法情况。
  
  如果原值与目标值相等,则什么也做不了,一改就不满足要求,不作为合法情况考虑。
  
  若原值小于目标值,则考虑(2),枚举所有能转移到\(i\)的偶位置\(j=i*p\)(其中\(p\)是枚举的质数),如果\(a_j \ge target-a_i\),那么合法情况就多了一种,因为\(j\)可以选\(target-a_i\)个数通过唯一一种方式——除去\(p\)——来到达\(i\)
  
​   那么概率也就很好计算了。
  
  
  

Code

  

#include <cstdio>
using namespace std;
const int N=1000005,MOD=998244353;
int n,a[N];
bool vis[N];
int p[N],pcnt,b[N],c[N];
void sieve(){
    int up=1e6;
    for(int i=2;i<=up;i++){
        if(!vis[i]){
            p[++pcnt]=i;
            b[i]=c[i]=1;
        }
        for(int j=1;j<=pcnt&&i*p[j]<=up;j++){
            int x=i*p[j];
            vis[x]=true;
            c[x]=c[i]^1;
            if(i%p[j]==0){
                b[x]=b[i];
                break;
            }
            b[x]=b[i]+1;
        }
    }
}
void readData(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",a+i);
}
inline int fmi(int x,int y){
    int res=1;
    for(;y;x=1LL*x*x%MOD,y>>=1)
        if(y&1) res=1LL*res*x%MOD;
    return res;
}
void solve(){
    int x=0;
    for(int i=1;i<=n;i++) 
        if(c[i]) x^=a[i];
    int legal=0;
    for(int i=1;i<=n;i++)
        if(c[i]){
            int best=x^a[i];
            if(best<a[i]) legal+=b[i];
            else{
                int delta=best-a[i];
                if(!delta) continue;
                for(int j=1;j<=pcnt&&i*p[j]<=n;j++) 
                    if(a[i*p[j]]>=delta) legal++;
            }
        }
    int all=0;
    for(int i=1;i<=n;i++) 
        (all+=1LL*a[i]*b[i]%MOD)%=MOD;
    int ans=1LL*legal*fmi(all,MOD-2)%MOD;
    printf("%d\n",ans<0?ans+MOD:ans);
}
int main(){
    sieve();
    readData();
    solve();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/RogerDTZ/p/9439644.html