有一個同學在學習分式。他需要將一個連分數化成最簡分數,你能幫助他嗎?
連分數是形如上圖的分式。在本題中,所有係數都是大於等於0的整數。
輸入的cont代表連分數的係數(cont[0]代表上圖的a0,以此類推)。返回一個長度爲2的數組[n, m],使得連分數的值等於n / m,且n, m最大公約數爲1。
示例 1:
輸入:cont = [3, 2, 0, 2]
輸出:[13, 4]
解釋:原連分數等價於3 + (1 / (2 + (1 / (0 + 1 / 2))))。注意[26, 8], [-13, -4]都不是正確答案。
示例 2:
輸入:cont = [0, 0, 3]
輸出:[3, 1]
解釋:如果答案是整數,令分母爲1即可。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/deep-dark-fraction
競賽只准備作簡單題,結果這個就死活通不過,現在看原來是漏掉了len==1的case。。。
class Solution {
public:
vector<int> fraction(vector<int>& cont) {
vector<int> res(2);
int len = cont.size();
if(len==1){ //特殊判斷,比賽的時候這個case一直想不到,又不報錯,心累哦0.0
res[1]=1;
res[0]=cont[0];
return res;
}
for(int i=len-1;i>0;--i){
if(i==len-1){ //倒數第一對,分母是最後一個數字,分子是最後一個數字乘倒數第二個數字
res[1]=cont[i];
res[0]=cont[i-1]*res[1]+1;
}else{ //之後的每一個,分母是之前的分子,分子是之前的分子乘現在的a,最後加上之前的分母;可能畫圖比較好理解
int temp = res[1];
res[1]=res[0];
res[0]=cont[i-1]*res[1]+temp;
}
}
return res;
}
};