Submit: 497 Solved: 326
[Submit][Status][Discuss]
Description
Input
Output
Sample Input
0 0 1 1
Sample Output
HINT
Source
感觉自己还是突破不了对省选题的恐惧心理
这题50分(实际80)暴力其实是很好想的
但是因为自己没怎么学过期望,所以一看到期望这两字就直接把这题弃了
思路:
对于每个灯来说,操作两次的结果与不操作相同的
因此我们可以预处理出初始局面需要的操作次数
约数的话直接$O(nlogn)$vector暴力求
这样的话最暴力的想法就是从大的向小的依次枚举,实际这就是最优策略
用$dp[i]$表示对于$n$盏灯,从需要按$i$次能全部熄灭到按$i-1$次能全部熄灭的期望
考虑这一次的情况
有$\frac{i}{n}$的概率按到需要按的灯,此时的期望为$\frac{i}{n}*1$
有$\frac{n-i}{n}$的概率按到不需要按的灯,此时的期望为$\frac{n-i}{n}*(dp[i]+dp[i+1]+1)$
那么$dp[i]=\frac{i}{n}+\frac{n-i}{n}*(dp[i]+dp[i+1]+1)$
化简一下
$dp[i]=\frac{n}{i}+\frac{(n-i)dp[i+1]}{i}$
这样就可以线性递推了
注意推的时候从$n$推就可以,因为最优状态下需要的操作次数一定小于等于$n$
答案为$n!\sum_{i=1}^{need}$
#include<cstdio> #include<cstring> #include<algorithm> #include<vector> #define int long long using namespace std; const int MAXN=1e6+10; const int mod=100003; //#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,MAXN,stdin),p1==p2)?EOF:*p1++) //int buf[1<<20],*p1=buf,*p2=buf; inline int read() { char c=getchar();int x=0,f=1; while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();} while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();} return x*f; } int N,k; int a[MAXN]; vector<int>Yue[MAXN]; int need; int dp[MAXN],inv[MAXN]; int fastpow(int a,int p) { int base=1; while(p) { if(p&1) base=(base*a)%mod; a=(a*a)%mod; p>>=1; } return base%mod; } main() { #ifdef WIN32 freopen("a.in","r",stdin); #endif N=read();k=read(); inv[1]=1; for(int i=2;i<=N;i++) inv[i] = ((-(mod/i)*inv[mod%i]%mod)+mod)%mod; for(int i=1;i<=N;i++) a[i]=read(); for(int i=1;i<=N;i++) for(int j=i;j<=N;j+=i) Yue[j].push_back(i); for(int i=N;i>=1;i--) if(a[i]) { for(int j=0;j<Yue[i].size();j++) a[Yue[i][j]]^=1; need++; } dp[N]=1; for(int i=N-1;i>k;i--) dp[i]=(N+(N-i)*dp[i+1])*inv[i]%mod; for(int i=k;i>=1;i--) dp[i]=1; int ans=0; for(int i=1;i<=need;i++) ans=(ans+dp[i])%mod; for(int i=1;i<=N;i++) ans=(ans*i)%mod; printf("%lld",ans); return 0; }