P3750 [六省联考2017]分手是祝愿（期望DP+质因数分解）

洛谷题目传送门
先从这个题跳出来，思考如何用最少的次数将灯关掉
因为一个灯能控制的灯都小于他，所以最右边的灯一定要按，因为如果按他的倍数就更不优，所以就从高到底扫一遍，遇到亮的灯就点一次，因为每个灯都是不可替代的，所以这启示我们这些灯在方案中是必须要按的
求出这些灯的个数$cnt$，也就是最少要按的次数，如果cnt小于等于k，则直接输出cnt即可
所以设$f[i]$表示将剩下$i$盏必须要按的灯变成（i-1）盏必须要按的灯的期望次数
$$f[i]=\frac{i}{n}\times 1+\frac{n-i}{n}\times (f[i]+f[i+1]+1)$$
改式的意义为，有$\frac{i}{n}$的概率按对，也就是将$i$盏必须要按的灯变为$i-1$盏需要按的灯，有$\frac{n-i}{n}$的概率按错，回到$i+1$盏灯的情况，然后还需要再按$f[i]$次按回来，最后的1是这一次的贡献
化简得
$f[i]=\frac{1}{i}\times (n+(n-i)\times f[i+1])$
最后答案就是$f[cnt]+f[cnt-1]+\dots \dots \dots \dots +f[k+1]+k$
也就是将cnt盏灯变为cnt-1盏灯，一直变到k盏灯最后按k次按完

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e5+7;
const int mod = 1e5+3;
int n,k,cnt=0;
int a[N];
vector<int> fac[N];
LL inv[N],F[N],ans;
LL Pow(LL a,int b)
{
	LL res=1;
	while(b)
	{
		if(b&1) res=res*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return res%mod;
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=n;j+=i)
		fac[j].push_back(i);
	}
	for(int i=n;i>=1;i--)
	{
		if(a[i]==1)
		{
			for(int p=0;p<fac[i].size();p++)
			{
				int j=fac[i][p];
				a[j]^=1;
			}
			cnt++;
		}
	}
	if(cnt<=k) ans=cnt%mod;
	else
	{
		for(int i=1;i<=n;i++)
		inv[i]=Pow(i,mod-2)%mod;
		F[n]=1;
		for(int i=n-1;i>=1;i--)
		F[i]=(i+(n-i)*(1+F[i+1]))*inv[i]%mod;
		for(int i=cnt;i>k;i--)
		ans=(ans+F[i])%mod;
		ans=(ans+k)%mod;		
	}
	for(int i=1;i<=n;i++)
	ans=(ans*i)%mod;
	cout<<ans;
	return 0;
}