集训 sum

来源:互联网 发布:js从入门到精通 编辑:程序博客网 时间:2024/06/04 18:16

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 longusing 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);}
原创粉丝点击