首先考虑,给你一个局面最少操作多少次,显然要从大往小按,可以证明这样是最优的。把这些按下的位置标记出来,可以证明一定要恰好按这些位置,别的不能动。因此问题转化为,给你一个序列,有若干位置需要被访问奇数次,后接原题意。这样注意到位置是无关的,只需要知道有几个位置是“需要被按下的”即可。然后设dp[x]表示这件事情,状态转移方程如下:
dp[x]=xn(dp[x−1]+1)+n−xn(dp[x+1]+1)
d
p
[
x
]
=
x
n
(
d
p
[
x
−
1
]
+
1
)
+
n
−
x
n
(
d
p
[
x
+
1
]
+
1
)
但是并不需要高斯消元,从dp[n]的转移中看出,dp[n-1]可以被唯一的表示成关于dp[n]的形式,再从dp[n-1]的转移中把dp[n-1]替换掉就能够把dp[n-2]写成关于dp[n]的形式……一路递推下去,直到dp[k]是已知的,就可以解出dp[n],再一路递推回来即可。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define N 100010
#define lint long long
#define mod 100003
using namespace std;
int a[N],b[N],c[N],d[N],s[N],f[N],v[N];
inline int fast_pow(int x,int k)
{
int ans=1;
while(k)
{
if(k&1) ans=(lint)ans*x%mod;
x=(lint)x*x%mod,k>>=1;
}
return ans;
}
int fac[N];
inline int inv(int x)
{ return fast_pow(x,mod-2); }
int main()
{
int n,k,ninv,x=0;scanf("%d%d",&n,&k),ninv=inv(n);
for(int i=1;i<=n;i++) scanf("%d",&v[i]);
for(int i=n;i>=1;i--)
{
for(int j=2*i;j<=n;j+=i) v[i]^=s[j];
if(v[i]) s[i]=1,x++;
}
a[n]=1,b[n]=0,c[n]=1,d[n]=1;
for(int i=n-1,ccinv;i>k;i--)
a[i]=(lint)i*ninv%mod,b[i]=(lint)ninv*(n-i)%mod,
ccinv=inv((1-(lint)b[i]*c[i+1]%mod+mod)%mod),
c[i]=(lint)a[i]*ccinv%mod,
d[i]=((lint)b[i]*d[i+1]%mod+1)%mod*ccinv%mod;
for(int i=0;i<=k;i++) f[i]=i;
for(int i=k+1;i<=n;i++)
f[i]=((lint)c[i]*f[i-1]%mod+d[i])%mod;
for(int i=fac[0]=1;i<=n;i++) fac[i]=(lint)fac[i-1]*i%mod;
return !printf("%d\n",(lint)f[x]*fac[n]%mod);
}