集训 sum

Description

N个数排成一个环，请选出不超过K段的连续的数，段与段间不能重叠，且使得选出的数和最大。

Input

输入文件sum.in第一行包含两个正整数N和k。
接下来1行描述这N个数。

Output

输出文件sum.out包含一个数，即要去的最大的和。

Sample Input

9 2
2 -1 2 -1 2 -4 1 -1 2
Sample Output

7
做法：这种查找。。线段树线段树线段树！！！

#include <cstdio>
#include <cstring>
#include <iostream>
#define ll long long
using namespace std;
int n,a[2000000];
ll t[20000000], ans, ma;

int read()
{
    int res = 0;
    char ch = getchar();
    while (ch < '0' || ch > '9')    ch = getchar();
    while (ch >= '0' && ch <= '9')      res = res * 10 + ch - '0', ch = getchar();
    return res;  
}

ll max(ll a, ll b)
{
    return a > b ? a : b;
}

void build(int p, int l, int r, int x)
{
    if (l == r && l == x)
    {
        t[p] = 1;
        return;
    }
    else if (l == r)    return;
    ll m = (l + r) >> 1;
    if (x <= m) build(p << 1, l, m, x);
        else    build(p << 1 | 1, m + 1, r, x);
    t[p] = t[p << 1] + t[p << 1 | 1];
} 

ll find(int p, int a, int b, int l, int r)
{
    if (a == l && b == r)   return t[p];
    else if (l == r)    return 0;
    int m = (l + r) >> 1;
    if (a > m)  return find(p << 1 | 1, a, b, m + 1, r);
        else if (b <= m)    return find(p << 1, a, b, l, m);
            else return find(p << 1, a, m, l, m) + find(p << 1 | 1, m + 1, b, m + 1, r);
}

int main()
{
    n = read();
    for (int i = 1; i <= n; i++)
    {
        a[i] = read();
        ma = max(ma, a[i]);
    }
    for (int i = 1; i <= n; i++)
    {
        ans += find(1, a[i] / 2 + 1, 10 * n, 1, 10 * n);
        build(1, 1, 10 * n, a[i]);
    }
    printf("%lld",ans);
}