拼多多笔试第三题回忆版:
有n只袜子,n为偶数,每只袜子的色号用一个正整数c表示,0<c<99;色差用d表示,0<d<50。如果随机抓取的两只袜子色号之差,小于色差d,表示可以配做一对袜子穿。现在给一组数代表袜子,和一个数代表色差,问随机抓取两只袜子能配成一对袜子的概率是多少?
例子:第一行输入 {25,5,1,70,30,50}
第二行输入 10
输出 13.333%
解法思路,先写一个C(m,n)的组合数的函数,求全部可能情况。然后用vector<> s存放从字符串中提取出来的数字,调用sort()对s排序,然后用for循环遍历s,用s[i]后面s[i+1]到s.end的每一个元素都去减s[i],判断差是否<=d,如果小于,计数器f就自增1。(这样做比较符合组合数的通常遍历思维,但应该还有能降低复杂度的方法,我暂时没想出来。)最后f就统计出了所有符合的组合的个数,再调用组合数函数combination(),强制转换f与combination的返回值为float类型,两个数相除就可以计算概率了。combination()的思路是时间复杂度较大的递归,思路很简单。
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int combination(int m,int n)
{
int sum=0;
if (m==n||n==0){
return 1;
}else {
return combination(m-1,n)+combination(m-1,n-1);
}
}
int main(){
vector<int> s;
int d=0,e=0,f=0,j=0,h=0,temp=0;
float g=0,k=0;
char c;
while(cin>>c){
if(c>='0'&&c<='9'){
temp=temp*10;
temp=temp+(c-'0');
} else {if(temp!=0){
s.push_back(temp);
temp=0;}
if(c=='}'){
break;
}
}
}
cin>>d;
sort(s.begin(),s.end());
for(int i=0;i<s.size()-1;i++){
j=i+1;
while((s[j]-s[i])<=d){
f++;
j++;
}
}
e=s.size();
h=combination(e,2);
g=static_cast <float>(f);
k=static_cast <float>(h);
g=(g/k)*100;
cout<<g<<"%";
return 0;
}