做题记录 Newcoder kotori和糖果 （记忆化搜索、递归

题目描述

链接：https://ac.nowcoder.com/acm/contest/21763/1022
来源：牛客网

kotori共有n块糖果，每块糖果的初始状态是分散的，她想把这些糖果聚在一堆。但她每次只能把两堆糖果合并成一堆。

已知把两堆数量为a和b的糖果聚在一堆的代价是|a-b|。

kotori想知道，她把这n块糖果聚在一堆的最小代价是多少？

题意

思路

• 最开始想按照merge的顺序依照题意模拟，但太难处理，而且也复杂度太大
• 意识到相等的两部分合并 或是 相差1的两部分合并就是花费最少的，所以可以递推。
• 提前递推出1e6内的答案，大于的递归求解

code

#include<bits/stdc++.h>
// #define DEBUG(x) (cout<<#x<<':'<<x<<' ')
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
// const int mod=1e9+7;
ll cnts[1000000];

ll dfs(ll n)
{
    if(n<1000000)return cnts[n];
    return (n&1)? dfs(n>>1)+dfs((n>>1)+1)+1 : dfs(n>>1)<<1;
}
void solv()
{
    ll n;
    cin>>n;
    if(n<1000000)cout<<cnts[n]<<'\n';
    else cout<<dfs(n)<<'\n';
}

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int T=1;
    cin>>T;
    for(int i=3;i<1000000;i++)
    {
        if(i&1)cnts[i]=cnts[i>>1]+cnts[(i>>1)+1]+1;
        else cnts[i]=cnts[i>>1]<<1;
    }
    while(T--)
    {
        solv();
    }
    return 0;
}

总结

知识补充