刷题记录:牛客NC50038kotori和糖果

传送门:牛客

题目描述:

kotori共有n块糖果，每块糖果的初始状态是分散的，她想把这些糖果聚在一堆。但她每次只能把两堆糖果合并成一堆。
已知把两堆数量为a和b的糖果聚在一堆的代价是|a-b|。
kotori想知道，她把这n块糖果聚在一堆的最小代价是多少？
输入:
2
5
6
输出:
2
2

这道题的贪心可能比较难想,即使想到了可能也会考虑会不会被T掉的风险,毕竟n最大为1e18

首先我是考虑用相同的数字一起合并的方法来考虑的,就是将每一个1先合并成2,再合并成4,以此类推,但是自己手推一下后发现似乎是没问题的,但是这么思考的话似乎很难使用代码来实现,因此我换了一种思路(主要是我看到了题目的标签中有递归与递推),我想了一下,最后一次合并的时候要么是相同的,要么是差一的话这样是最优的,并且两边的合并次数也恰好可以以之前的数字来推出来

然后我写了以下的代码:

map<ll,ll>s;
ll solve(ll n) {
	if(s[n]!=0) return s[n];
	else {
		if(n&1) {
			return s[n]=solve(n/2)+solve(n/2+1)+1;
		}else {
			return s[n]=solve(n/2)*2;
		}
	}
}
s[0]=0,s[1]=0,s[2]=0;

大家能发现上面的代码有什么问题吗,交到牛客上挂的地方显示的内存超限,如果有人和我打的是差
不多的代码的话应该会挺绝望的,首先刚开始我的代码是有预处理的,跑了300000组数据差不多,交上去内存超限之后我删掉了预处理的部分,然后还是超限,然后我把数组改成了map,然后还是超限,我当时直接自闭好吧,在比对了其他人的代码后发现和我代码的思路肯定是一致的,也就是说我的代码思路肯定是没问题的.但是是哪里挂了呢,我仔细的看了一下,TMD是s[n]!=0挂了,因为s[0]=s[1]=s[2]=0,虽然他们是0但是不代表并没有赋值,不知道有没有人被这个小bug给困扰住了呢??

下面是具体的代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
ll T,n;
map<ll,ll>s;
ll solve(ll n) {
	if(s[n]!=0||n<=2) return s[n];
	else {
		if(n&1) {
			return s[n]=solve(n/2)+solve(n/2+1)+1;
		}else {
			return s[n]=solve(n/2)*2;
		}
	}
}
int main() {
	s.clear();
	s[0]=0,s[1]=0,s[2]=0;
	cin>>T;
	while(T--) {
		cin>>n;
		cout<<solve(n)<<endl;
	}
	return 0;
}