题目描述
在你的面前从左到右摆放着n根长短不一的木棍,你每次可以折断一根木棍,并将折断后得到的两根木棍一左一右放在原来的位置(即若原木棍有左邻居,则两根新木棍必须放在左邻居的右边,若原木棍有右邻居,新木棍必须放在右邻居的左边,所有木棍保持左右排列)。折断后的两根木棍的长度必须为整数,且它们之和等于折断前的木棍长度。你希望最终从左到右的木棍长度单调不减,那么你需要折断多少次呢?
用例
输入:[3,5,13,9,12]
输出:1
输入:[3,12,13,9,12]
输出:2
输入:[3,13,12,9,12]
输出:3
输入:[3,13,60,7]
输出:10
输入:[3,63,7]
输出:8
输入:[9,1]
输出:8
单调栈
int breakNum(vector<int>& nums)
{
int ans = 0;
stack<int> st;
for (int i = nums.size() - 1; i >= 0; i--)
{
if (!st.empty() && st.top() < nums[i])
{
int t = (nums[i] - 1) / st.top();
ans += t;
st.push(nums[i] / (t + 1));
continue;
}
st.push(nums[i]);
}
return ans;
}
优化一下只用单调栈的思想
int breakNum(vector<int>& nums)
{
int ans = 0;
for (int i = nums.size() - 2; i >= 0; i--)
{
if (nums[i + 1] >= nums[i]) continue;
int t = (nums[i] - 1) / nums[i + 1];
ans += t;
nums[i] /= (t + 1);
}
return ans;
}