[2019杭电多校第三场][hdu6606]Distribution of books

dp好难呐

这道题用到的知识点是dp+线段树+二分答案。
简述一下解体流程。
首先地看到求最小最大解想到用二分答案。
二分答案需要验证是否可行，验证主要就是看能不能把一堆书以一个x为上限分出来k堆，如果这个x做不到分成k堆，那么肯定不行。
再看如何验证，定义一个dp[i]，i表示当前是第i本书，存的是前i本能分成的最大堆数。那么转移就是
if（sum[i] - sum[j - 1 ] >= x）dp[i] = dp[j-1]+1；
到此为止已经处理除了朴素的解。
但是朴素的解很显然会T，接下来处理优化。下面是N2算法的代码。

///注意这个解会T
#include<bits/stdc++.h>

using namespace std;
const int maxn = 2e5 + 10;
int num[maxn];
int sum[maxn];
int dp[maxn];
int n, k;

bool check(int mid) {
    bool f = false;
    memset(dp, 0, sizeof(dp));
    for (int i = 1; i <= n; ++i) {
        for (int j = 0; j < i; ++j) {
            if (sum[i] - sum[j] <= mid)
            {
                dp[i] = max(dp[i],dp[j] + 1);
                if (dp[i] == k) f = true;
            }
        }
    }
    return f;
}

int main() {
//    freopen("in.txt", "r", stdin);
    int T;
    cin >> T;
    while (T--) {
        int ans =0 ;
        cin >> n >> k;
        for (int i = 1; i <= n; ++i) {
            cin >> num[i];
            sum[i] = sum[i - 1] + num[i];
        }
        int l = -1e9;
        int r = 1e9;
        while (l <= r) {
            int mid = (l + r) / 2;
            if (check(mid)) {
                ans = mid;
                r = mid - 1;
            } else
                l = mid + 1;
        }
        cout << ans << endl;

    }
}

先看上面那个式子由于我们每次转移都是+1,那么我们可以直接在满足if（sum[i] - sum[j - 1 ] >= x）的范围内找最大的dp[i]，再拿出来转移+1就可以了。这里可以拿线段树存一下处理最大值。
AC代码

#include<bits/stdc++.h>

using namespace std;

#define int long long
const int maxn = 5e5 + 10;
int num[maxn];
int sum[maxn];
int cp[maxn];
int dp[maxn];
int n, k;
int len;
int tree[maxn<<2];

void push_up(int i)
{
    tree[i] = max(tree[i<<1],tree[(i<<1)|1]);
}
void build(int l,int r,int root)
{
    tree[root] = -1e9;
    if (l == r) return;
    int mid = (l+r)>>1;
    build(l,mid,root<<1);
    build(mid+1,r,root<<1|1);
}
void update(int pos,int val,int l,int r,int root)
{
    if (l == r)
    {
        tree[root] = val;
        return;
    }
    int mid = (l+r)>>1;
    if (pos <= mid)
        update(pos,val,l,mid,root<<1);
    else
        update(pos,val,mid+1,r,root<<1|1);
    push_up(root);
}
int query(int L,int R,int l,int r,int root)
{

    if (L > R) return -1e9;
    if (L <= l && R >= r) return tree[root];
    int ans = -1e9;
    int mid = (l + r) >> 1;
    if (L <= mid) ans = max(ans, query(L, R, l,mid,root<<1));
    if (R >= mid) ans = max(ans, query(L, R,mid+1,r,root<<1|1));
    return ans;
}

bool check(int mid) {
    int f = false;
    build(1,len,1);
    memset(dp,0,sizeof(dp));
    for (int i = 1; i <= n; ++i) {
        int pos = lower_bound(cp+1,cp+1+len,sum[i]-mid) - cp;
        int pos_end = lower_bound(cp+1,cp+1+len,sum[i]) - cp;

        dp[i] = query(pos,len,1,len,1)+1;
        if (sum[i] <= mid)
        {
            dp[i] = max(dp[i],1ll);
        }
        if (dp[i] >= k) {
            f = true;
        }
        if (dp[i] > 0 ) update(pos_end,dp[i],1,len,1);
    }
    return f;
}

signed main() {
//    freopen("in.txt", "r", stdin);
    int T;
    cin >> T;
    while (T--) {
        int ans =0 ;
        cin >> n >> k;
        for (int i = 1; i <= n; ++i) {
            cin >> num[i];
            sum[i] = sum[i - 1] + num[i];
            cp[i] = sum[i];
        }
        sort(cp+1,cp+1+n);
        len = unique(cp+1,cp+1+n)-cp-1;
        int l = -1e10;
        int r = 1e10;
        while (l <= r) {
            int mid = (l + r) / 2;
            if (check(mid)) {
                ans = mid;
                r = mid - 1;
            } else
                l = mid + 1;
        }
        cout << ans << endl;

    }
}