[leetcode/lintcode 题解] 阿里面试真题:双色塔

简介: [leetcode/lintcode 题解] 阿里面试真题:双色塔

描述
现在有红,绿两种颜色的石头,现在我们需要用这两种石头搭建一个塔,塔需要满足如下三个条件:

  1. 第1层应该包含1块石头,第2层应该包含2块,第i层需要包含i块石头。
  2. 同一层的石头应该是同一个颜色(红或绿)。
  3. 塔的层数尽可能多。

在满足上面三个条件的前提下,有多少种不同的建造塔的方案?当塔中任意一个对应位置的石头颜色不同,我们就认为这两个方案不相同。石头可以不用完。
由于答案可能会很大,请对10^9+7取模。

  • red+green≥1
  • 0≤red,green≤6×104

在线评测地址:领扣题库官网

样例1
输入: 4 6
输出: 2
说明: 有两种方案:[红,绿,红,绿],[绿,绿,绿,红]
AI 代码解读

源代码

public int twoColorsTower(int red, int green) {
        if (red == 0 || green == 0) {
            return 1;
        }

        if (red > green) {
            int temp = red;
            red = green;
            green = temp;
        }
        // dp[i&1][j]表示前i层一共放j个红石头
        int[][] dp = new int[2][red + 1];
        dp[1][0] = dp[1][1] = 1;
        int level = (int) Math.sqrt(2 * (red + green));
        int sum = 1, lower = 0, upper = red;
        int curr = 1, prev;
        int MOD = (int) 1.0e9 + 7;
        
        for (int i = 2; i <= level; i++) {
            sum += i;
            int tmpUpper = Math.min(sum, red);
            int tmpLower = Math.max(sum - green, 0);
            // 红石头不够了,已经是最高层,停止更新
            if (tmpLower > tmpUpper) break;
            upper = tmpUpper;
            lower = tmpLower;
            prev = curr;
            curr = curr ^ 1;

            // j小于本层i,红石只能是之前都放完了
            for (int j = lower; j < i; j++) {
                dp[curr][j] = dp[prev][j];
            }
            // 转移方程:dp[i][j] = dp[i-1][j] + dp[i-1][j-i] (when j>=i)
            // dp[i-1][j]表示第i层放绿石 dp[i-1][j-i]表示i层放红石
            for (int j = i; j <= upper; j++) {
                dp[curr][j] = (dp[prev][j] + dp[prev][j - i]) % MOD;
            }
        }
        int ans = 0;
        for (int j = lower; j <= upper; j++) {
            ans = (ans + dp[curr][j]) % MOD;
        }
        return ans;
    }
AI 代码解读

更多题解参考:九章官网solution

目录
打赏
0
0
0
0
6
分享
相关文章
阿里腾讯互联网公司校招 Java 面试题总结及答案解析
本文总结了阿里巴巴和腾讯等互联网大厂的Java校招面试题及答案,涵盖Java基础、多线程、集合框架、数据库、Spring与MyBatis框架等内容。从数据类型、面向对象特性到异常处理,从线程安全到SQL优化,再到IOC原理与MyBatis结果封装,全面梳理常见考点。通过详细解析,帮助求职者系统掌握Java核心知识,为校招做好充分准备。资源链接:[点击下载](https://pan.quark.cn/s/14fcf913bae6)。
57 2
【IO面试题 四】、介绍一下Java的序列化与反序列化
Java的序列化与反序列化允许对象通过实现Serializable接口转换成字节序列并存储或传输,之后可以通过ObjectInputStream和ObjectOutputStream的方法将这些字节序列恢复成对象。
面试官:单核 CPU 支持 Java 多线程吗?为什么?被问懵了!
本文介绍了多线程环境下的几个关键概念,包括时间片、超线程、上下文切换及其影响因素,以及线程调度的两种方式——抢占式调度和协同式调度。文章还讨论了减少上下文切换次数以提高多线程程序效率的方法,如无锁并发编程、使用CAS算法等,并提出了合理的线程数量配置策略,以平衡CPU利用率和线程切换开销。
面试官:单核 CPU 支持 Java 多线程吗?为什么?被问懵了!
大厂面试高频:什么是自旋锁?Java 实现自旋锁的原理?
本文详解自旋锁的概念、优缺点、使用场景及Java实现。关注【mikechen的互联网架构】,10年+BAT架构经验倾囊相授。
大厂面试高频:什么是自旋锁?Java 实现自旋锁的原理?
大厂面试必看!Java基本数据类型和包装类的那些坑
本文介绍了Java中的基本数据类型和包装类,包括整数类型、浮点数类型、字符类型和布尔类型。详细讲解了每种类型的特性和应用场景,并探讨了包装类的引入原因、装箱与拆箱机制以及缓存机制。最后总结了面试中常见的相关考点,帮助读者更好地理解和应对面试中的问题。
204 4
探讨面试常见问题雪花算法、时钟回拨问题,java中优雅的实现方式
【10月更文挑战第2天】在大数据量系统中,分布式ID生成是一个关键问题。为了保证在分布式环境下生成的ID唯一、有序且高效,业界提出了多种解决方案,其中雪花算法(Snowflake Algorithm)是一种广泛应用的分布式ID生成算法。本文将详细介绍雪花算法的原理、实现及其处理时钟回拨问题的方法,并提供Java代码示例。
500 2

热门文章

最新文章

AI助理
登录插画

登录以查看您的控制台资源

管理云资源
状态一览
快捷访问

你好,我是AI助理

可以解答问题、推荐解决方案等