BZOJ4872 [Shoi2017]分手是祝愿

最新推荐文章于 2021-10-27 18:41:59 发布

neither_nor

最新推荐文章于 2021-10-27 18:41:59 发布

阅读量1.4k

点赞数 1

分类专栏： BZOJ DP、递推

本文链接：https://blog.csdn.net/neither_nor/article/details/70598175

版权

BZOJ 同时被 2 个专栏收录

382 篇文章 1 订阅

订阅专栏

DP、递推

84 篇文章 1 订阅

订阅专栏

YY一下给定一个局面怎么求最少步数，就是可以转换成一个异或方程组，然后令所有变量的和最小

我们可以发现把这个异或方程组用矩阵表示出来之后，他的对角线是1，而下三角里全是0……

也就是说不需要消元，并且有唯一解

那么我们还知道这个矩阵的结构比较特殊（只有在i的约数行第i列才为1），那么我们可以直接预处理每个数的所有约数然后爆解，由调和级数知解出来的复杂度是n log n的

那么考虑一下加入了当答案>k时候随机走的条件，我们发现一种局面的期望步数只与还有多少个灯需要被按一次（即当前情况下的最少步数有关）

设还有i个灯需要被按的时候的期望步数为ans[i]，那么我们发现ans[i]只与ans[i-1]和ans[i+1]有关

ans[i]=(i/n)*ans[i-1]+((n-i)/n)*ans[i+1]+1

并且ans[n]=ans[n-1]+1，那么我们可以用ans[i]表示出ans[i+1]，进而用ans[i-1]表示出ans[i]……

到i等于k的时候，因为直接走最少步数，所以ans[k]就等于k，然后再推回去就可以了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<iomanip>
#include<ctime>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<bitset>
#include<set>
using namespace std;
#define MAXN 100010
#define MAXM 1010
#define eps 1e-8
#define ll long long 
#define MOD 100003
#define INF 1000000000
vector<int>ps[MAXN];
int a[MAXN];
ll ans;
int n,k;
ll f[MAXN],g[MAXN];
int mi(ll x,int y){
	ll re=1;
	while(y){
		if(y&1){
			(re*=x)%=MOD;
		}
		(x*=x)%=MOD;
		y>>=1;
	}
	return re;
}
void main1(){
	int i,j;
	for(i=1;i<=n;i++){
		for(j=i;j<=n;j+=i){
			ps[j].push_back(i);
		}
	}
	for(i=n;i;i--){
		if(a[i]){
			ans++;
			for(j=0;j<ps[i].size();j++){
				a[ps[i][j]]^=1;
			}
		}
	}
	if(ans<=k){
		for(i=1;i<=n;i++){
			(ans*=i)%=MOD;
		}
		printf("%lld\n",ans);
		return ;
	}
	f[n]=g[n]=1;
	ll nin=mi(n,MOD-2);
	for(i=n-1;i>k;i--){
		ll t=mi((1-(n-i)*nin%MOD*f[i+1]%MOD+MOD)%MOD,MOD-2);
		f[i]=i*nin%MOD*t%MOD;
		g[i]=((n-i)*nin%MOD*g[i+1]%MOD+1)*t%MOD;
	}
	ll t=k;
	for(i=k+1;i<=ans;i++){
		t=(f[i]*t+g[i])%MOD;
	}
	for(i=1;i<=n;i++){
		(t*=i)%=MOD;
	}
	printf("%lld\n",t);
}
int main(){
	int i;
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	main1();
	return 0;
}

/*

*/