腾讯笔试题-分治+线段树

题目:有n块木板,第i块木板的长度是ai,然后要对这些木块刷漆,一次可以刷连续的一行,或者一列,

输入一个n和n个木板的长度,比如:

n最大不超过5000

5
2 2 2 1 2
//木板分布
*** *
*****
输出:3
可以先刷下面一行,然后刷上面一行的左边3个,再刷右边一个

大体思路:

大体来说就是一个分治的思路,找到区间的最小值,然后刷最下面的部分,对于上面的部分再分开刷,当然,可能会有竖着刷效率更好的情况,这种情况就要竖着刷。

这个过程涉及到一个动态查询区间最小值的问题,5000个数字,最差情况下就会有5000次查询,所以查询就不能直接遍历,因此使用线段树做查询的处理。

好久没写过线段树了,调了快一个小时才刚调通,最后只AC了55%,以下是调整之后的代码,已经改正了一个问题,应该能够AC了。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node {
    int pos, lc, rc, b, e;
}tn[20000];
long long id[5005];
int no = 0;
int init(int u, int b, int e) {
    tn[u].b = b;
    tn[u].e = e;
    if(b == e - 1) {
        tn[u].pos = b;
        return b;
    }
    tn[u].lc = ++no;
    tn[u].rc = ++no;
    int m = (b + e) / 2;
    int x = init(tn[u].lc, b, m);
    int y = init(tn[u].rc, m, e);
    if(id[x] < id[y])
        tn[u].pos = x;
    else tn[u].pos = y;
    return tn[u].pos;
}
int quary(int u, int b, int e) {
    // printf("%d %d %d\n", u, b, e);
    if(b >= e) {
        printf("zhe li cuo le\n");
        exit(0);
    }
    if(b >= tn[u].e || e <= tn[u].b) {
        printf("must be wrong\n");
        exit(0);
        return -1;
    }
    if(b == tn[u].b && e == tn[u].e)
        return tn[u].pos;
    int m = (tn[u].b + tn[u].e) / 2;
    int x = -1, y = -1;
    if(b < m)
        x = quary(tn[u].lc, b, min(m, e));
    if(m < e) {
        y = quary(tn[u].rc, max(m, b), e);
    }
    if(x == -1 || y == -1) return x + y + 1;
    if(id[x] < id[y]) return x;
    return y;
}
long long cul(int m, int b, int e) {

    if(e - b <= 0) return 0;
    int x = quary(0, b, e);
    // printf("QQQ   %d %d %d %d\n", m, b, e);
    int ans = id[x] - m;
    if(x < e)
        ans += cul(id[x], b, x);
    ans += cul(id[x], x + 1, e);
    return min(ans, e - b);
}
int main() {
    memset(tn, 0, sizeof(tn));
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        scanf("%d", &id[i]);
    }
    init(0, 0, n);
    // printf("%d\n", quary(0, 0, n - 1));
    // return 0;

    printf("%lld\n", cul(0, 0, n));
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值