京东笔试之石头分堆

题目描述

小明得到了n个石头，他想把这些石头分成若干堆，每堆至少有一个石头。他把这些石堆排在一条直线上，他希望任意相邻两堆的石头数都不一样。小明最后的得分为石头数大于等于k的石堆数，问他最多能得多少分。
严格地，小明把n个石头分成了m堆，每堆个数依次为a1，a2，…..，am。要求满足：
1、ai≥1（1≤i≤m）
2、ai≠ai+1（1≤i＜m）
3、a1+a2+…+am＝n
小明想知道a1，a2…..，am中大于等于k的数最多能有多少个？
输入
输入两个数n, k。（1≤k≤n≤109）
输出
输出最大的得分

样例输入
5 1
样例输出
3
Hint
输入样例2
4 2
输出样例2
1
输入样例3
2 1
输出样例3
1

解题思路

第一次我的想法是k,1,k,1……这样的类型，但是只过了30%的用例，后来发现这样解法的问题，这样虽然满足了题目的条件，但是1就浪费了。
所以后来想了k,k+1,k,k+1…..才是最佳的

#include <iostream>
using namespace std;
int main()
{
    int n, k;
    int ans = 0;
    cin >> n >> k;
    ans = n / (2 * k + 1);
    ans *= 2;
    if (n % (2 * k + 1) >= k)
        ans = ans + 1;
    cout << ans;
    return 0;
}