hello-algo/ja/codes/java/chapter_greedy/max_capacity.java

/**
 * File: max_capacity.java
 * Created Time: 2023-07-21
 * Author: krahets ([email protected])
 */

package chapter_greedy;

public class max_capacity {
    /* 最大容量：貪欲法 */
    static int maxCapacity(int[] ht) {
        // i, j を初期化し、配列の両端をそれぞれ指すようにする
        int i = 0, j = ht.length - 1;
        // 最大容量の初期値は 0
        int res = 0;
        // 2 枚の板が出会うまで貪欲な選択を繰り返す
        while (i < j) {
            // 最大容量を更新する
            int cap = Math.min(ht[i], ht[j]) * (j - i);
            res = Math.max(res, cap);
            // 低い方の仕切り板を内側に移動する
            if (ht[i] < ht[j]) {
                i++;
            } else {
                j--;
            }
        }
        return res;
    }

    public static void main(String[] args) {
        int[] ht = { 3, 8, 5, 2, 7, 7, 3, 4 };

        // 貪欲アルゴリズム
        int res = maxCapacity(ht);
        System.out.println("最大容量は " + res);
    }
}