蓝桥杯2022砍竹子

P2040 - [蓝桥杯2022初赛] 砍竹子 - New Online Judgehttp://oj.ecustacm.cn/problem.php?id=2040

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N = 2e5 + 4;
vector<int>s[N];
ll cut(ll h) {
    return sqrt(h / 2 + 1);
}
int main() {
    ios::sync_with_stdio(false);
    int n, ans = 0;
    cin >> n;
    for (int i = 0; i < n; i++) {
        ll h;
        cin >> h;
        while (h > 1) {
            s[i].push_back(h);
            h = cut(h);
            ans++;
        }
    }
    for (int i = 1; i < n; i++) {
        for (int j = 0; j < s[i].size(); j++) {
            for (int k = 0; k < s[i - 1].size(); k++) {
                if (s[i][j] == s[i - 1][k]) {
                    ans--;
                }
            }
        }
    }
    cout << ans;
    return 0;
}

每读入一个竹子的高度h，就计算出单独把他砍到1需要的次数和每一次的高度。

遍历每个竹子经历的所有高度，和他前一个竹子经历的所有高度一一比较，如果有相同的，则次数可以-1