题目描述
小林在玩一个抽卡游戏,其中有 n 种不同的卡牌,编号为 1 到 n。每一次抽卡,她获得第 i 种卡牌的概率为 pi。如果这张卡牌之前已经获得过了,就会转化为一枚硬币。可以用 k 枚硬币交换一张没有获得过的卡。
小林会一直抽卡,直至集齐了所有种类的卡牌为止,求她的期望抽卡次数。如果你给出的答案与标准答案的绝对误差不超过 10−4,则视为正确。
提示:聪明的小林会把硬币攒在手里,等到通过兑换就可以获得剩余所有卡牌时,一次性兑换并停止抽卡。
输入格式
从标准输入读入数据。
输入共两行。第一行包含两个用空格分隔的正整数 n,k,第二行给出 p1,p2,…,pn,用空格分隔。
输出格式
输出到标准输出。
输出共一行,一个实数,即期望抽卡次数。
#include<bits/stdc++.h>
using namespace std;
double ans;
int m,n;
void dfs(int num,int now,double thisans,int coin,double *pi,int *mark,int turn){//num 表示 卡牌总种类 now 是 手头卡牌种类
// thisans 指当前回合概率 // coin 指硬币数量 *pi是抽到当前卡的概率集//mark 存0、1,表示是否有该种种类卡牌
if(now==num){
ans+=thisans*turn;
cout<<"ok the ans=="<<thisans*turn<<endl;
return;
}
else if(coin/n+now>=num) {
ans+=thisans*turn;
cout<<"the ans=="<<thisans*turn<<endl;
return;
}
// else if(turn >= num*n) return ;
for(int i=1;i<=num;i++){
if(mark[i]==0){//抽到不同的卡牌
mark[i]=1;
dfs(num,now+1,thisans*pi[i],coin,pi,mark,turn+1);
mark[i]=0;
}
else
dfs(num,now,thisans*pi[i],coin+1,pi,mark,turn+1);
//cout<<"dfs"<<endl;
}
}
int main(){
cin>>m>>n;
double *pi=new double [m+1];
int *mark=new int [m+1]; for(int i=1;i<=m;i++) mark[i]=0;
for(int i=1;i<=m;i++) cin>>pi[i];
dfs(m,0,1,0,pi,mark,0);
cout<<ans;
}
自己写了DFS 跑过了样例,但不知为啥是0分
有空会改改