[LeetCode]Candy 递归

题意:分发糖果,要求具有更大rating的人比他相邻的具有更小rating的人的糖果数多。求最少要分给这些人多少个。

rating:4,3,1,2,5

糖果:3,2,1,2,3

通过分析可知,比两个邻居ratings都小的人要分1个(包括了最左边的人和最右边的人),其余的人糖果数为邻居中比他rating小的人的糖果数+1。

因为在遍历时,可能分发的糖果数依赖于其他节点,所以可采用递归求解。

class Solution {
public:
    int* val,n;
    vector<int>*vt;
    int dfs(int u){
        if(u<0||u>=n)return 0;
        if(val[u]!=0)return val[u];
        if((u==0&&(*vt)[u]<(*vt)[u+1])||(u==n-1&&(*vt)[u]<(*vt)[u-1])||(u<n&&u>0&&(*vt)[u]<(*vt)[u+1]&&(*vt)[u]<(*vt)[u-1]))
            val[u]=1;
        else{
			if(u<n&&(*vt)[u]>(*vt)[u+1])
				val[u]=dfs(u+1);
			if(u>0&&(*vt)[u]>(*vt)[u-1])
				val[u]=max(val[u],dfs(u-1));
			val[u]++;
		}
        return val[u];
    }
    int candy(vector<int> &rt) {
        n=rt.size();
        if(n==1)return 1;
        val=new int[n];
        vt=&rt;
        memset(val,0,sizeof(int)*n);
        int i;
        for(i=0;i<n;++i){
            if(val[i]==0){
                dfs(i);
            }
        }
        int ans=0;
        for(i=0;i<n;++i)
            ans+=val[i];
        delete[]val;
        return ans;
    }
};


评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值