題目見圖,大致意思是n天計劃來減肥,每天分爲減肥a斤和增重b斤以及對應的概率,問n天后博主的體重的期望是多少。
思路:搜一下期望定義寫出最終所求的公式:最終體重*該體重對應的概率 求和即可。
採用樹的結構來實現這個公式。然後找到這個樹對應的所有路徑 再按照公式計算。
下面給出構建二叉樹的代碼,和找到所有路徑的代碼,按照公式計算沒寫因爲牽扯到分數類型的運算,我不知道標準庫裏面有沒有,難道還要自己寫一個分數類嗎?這筆試題真心有毒。
struct Tree{
int val;
int p,q;
Tree* left;
Tree* right;
Tree(int a,int p1,int q1):val(a),p(p1),q(q1),left(nullptr),right(nullptr){}
};
void conStructMy(Tree* root,vector<vector<int> >& arry,int i){
int a=0, b=0, p=0, q=0;
a = arry[i][0];
b = arry[i][1];//arry[i][1];
p = arry[i][2];//arry[i][2];
q = arry[i][3];//arry[i][3];
Tree* nodel = new Tree(-a,p,q);
Tree* noder = new Tree(b,q-p,q);
root->left = nodel;
root->right = noder;
if(i<4)
{
conStructMy(root->left, arry, i+1);
conStructMy(root->right, arry, i+1);
}
return;
}
void pathHelper(vector<vector<int>> &result,vector<int> &path,Tree* root)
{
if(root == nullptr)
return;
path.push_back(root->val);
if(root->left == nullptr && root->right == nullptr)
{
result.push_back(path);
path.pop_back();
return;
}
pathHelper(result, path, root->left);
pathHelper(result, path, root->right);
path.pop_back();
}
//全路徑搜索。
vector<vector<int>> allPath(Tree* root)
{
vector<vector<int>> result;
vector<int> path;
pathHelper(result, path, root);
for(int i = 0;i < result.size();i++)
{
for(auto x : result[i])
{
cout << x << " ";
}
cout << endl;
}
return result;
}
int main()
{
vector<vector<int> > arry {{1,5,1,2},{2,4,1,2},{2,4,1,4},{2,4,2,7},{1,1,1,1}};
Tree* root = new Tree(90,1,1);
// for(int i = 0;i < 5;i++)
// {
// for(int j = 0; j < 4;j++)
// cout << arry[i][j] << " ";
// cout << endl;
// }
conStructMy(root, arry, 0);
}
2018.8.1更新
來填個坑,大疆的線上筆試在賽碼網平臺,博主一直沒收到面試通知,去官網看了下發現筆試不通過0-0 不知道這家公司是不是不怎麼缺人。不但題出的巨坑且還不給過,不建議投這家公司,有點浪費時間咯。並不是因爲別的,只是上面的理由。