HDU - 3486 Interviewe （RMQ + 枚举）

HDU - 3486 Interviewe

题目

给你 n 个数，让你分成 m 组，每组长度 n / m，剩下的人不要。每组最大值之和要大于给出的 k，问最小的 m 是多少。

分析

看到分组取最值，很明显 RMQ 问题，这里最小的 m 肯定是 1了，那么就从 1 开始枚举，只要遇到满足条件的就直接输出。注意要加个剪枝：枚举 m 的时候有可能两个 m值对应的分组长度是相同的。这时就不用再次计算了，直接取上次的值加上一个区间最值即可。

代码

#include <bits/stdc++.h>
#define d(x) cout << (x) << endl;
using namespace std;
const int N = 2e5 + 10;

int n, k;
int a[N];
int st[N][20];

void rmq(){
    for(int i = 0; i < n; i++){
        st[i][0] = a[i];
    }
    for (int j = 1; (1 << j) <= n; j++){
        for(int i = 0; i + (1<<j) - 1 < n; i++){
            st[i][j] = max(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
        }
    }
}

int ask(int l, int r){
    int k = log2(r - l + 1);
    return max(st[l][k], st[r - (1 << k) + 1][k]);
}

int solve(){                            // 找尽可能小的 m
    for(int i = 2; i <= n; i++){         // 分成几组
        int len = n / i;
        int sum = 0, s = 0, e = len - 1, temp;
        if(i != 2 && n/i == n/(i-1)){
            s = (n / (i - 1)) * (i - 1);
            sum = temp + ask(s, s + len - 1);
        }else{
            int k = i;
            while(k--){
                sum += ask(s, e);
                s = e + 1;
                e += len;
            }
        }
        temp = sum;
        if(sum > k){
            return i;
        }
    }
    return -1;
}

int main()
{
    while(scanf("%d%d", &n, &k)){
        if(n < 0)
            break;
        int maxn = 0, sum = 0;
        for(int i = 0; i < n; i++){
            scanf("%d", &a[i]);
            sum += a[i];
            maxn = max(maxn, a[i]);
        }
        if(sum <= k){               // 分成 n 组都超不过 k
            printf("-1\n");
            continue;
        }
        if(maxn > k){               // 一组就可以达成要求
            printf("1\n");
            continue;
        }
        rmq();
        printf("%d\n", solve());
    }
    return 0;
}