第十三届蓝桥杯C++B组省赛 J 题——砍竹子

题目描述
这天，小明在砍竹子，他面前有 n 棵竹子排成一排，一开始第 i 棵竹子的高度为 hi.他觉得一棵一棵砍太慢了，决定使用魔法来砍竹子。魔法可以对连续的一段相同高度的竹子使用，假设这一段竹子的高度为H，那么使用一次魔法可以把这一段竹子的高度都变为：

其中 ⌊x⌋ 表示对 x 向下取整。

小明想知道他最少使用多少次魔法可以让所有的竹子的高度都变为1。

输入格式

第一行为一个正整数 n，表示竹子的棵数。
第二行共 n 个空格分开的正整数 hi，表示每棵竹子的高度。
20%的测试数据：n ≤ 1000; hi ≤ 10^6。
100%的测试数据：n ≤ 2 × 10^5; hi ≤ 10^18。

输出格式

一个整数表示答案。

输入样例
6
2 1 4 2 6 7

输出样例
5

把一棵树砍掉所需要的次数暴力求和累加起来，然后再减掉相邻相等的树；

Java代码：

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Main {
    static BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
    static PrintWriter out=new PrintWriter(new OutputStreamWriter(System.out));
    static Map<Integer,List<Long>> map=new HashMap<>();
    public static void main(String[] args) throws IOException {
        int n=Integer.parseInt(br.readLine());
        String[] s = br.readLine().split(" ");
        int ans=0;
        for (int i=0;i<n;i++){
            long h = Long.parseLong(s[i]);
            while (h>1){
                add(i,h);
                h= (long) Math.sqrt(h/2+1);
                ans++;
            }
        }
        for (int i=1;i<n;i++){
            if (map.containsKey(i)&&map.containsKey(i-1)){
                for (Long x : map.get(i)) {
                    for (Long y : map.get(i - 1)) {
                        if (x==y) ans--;
                    }
                }
            }
        }
        out.println(ans);
        out.flush();
    }

    static void add(int a,long b){
        if (!map.containsKey(a)) map.put(a,new ArrayList<>());
        map.get(a).add(b);
    }
}