SPOJ 2916 Can you answer these queries V

SPOJ_2916

    这个题目需要分情况讨论一下，如果y1<x2，那么就是前面区间的一个后缀，加上后面区间的一个前缀，再加上中间的部分，如果y1>=x2，则根据位置的不同又可以另分三种情况，逐一讨论并更新最优解即可。

    分析之后就发现，在线段树的基础上写两个查询函数即可，一个就是和SPOJ_1716的GSS1那样功能的函数（在我的程序里面对应的是query这个函数），另一个是可以求一个区间最大的前缀和或者后缀和的函数（在我的程序里面对应的是Search这个函数）。

#include<stdio.h>
#include<string.h>
#define INF 0x3f3f3f3f
#define MAXD 10010
int N, A[MAXD], a[MAXD], mc[4 * MAXD], lc[4 * MAXD], rc[4 * MAXD];
int Max(int x, int y)
{
    return x > y ? x : y;
}
void update(int cur, int x, int y)
{
    int mid = (x + y) >> 1, ls = cur << 1, rs = cur << 1 | 1;
    mc[cur] = Max(mc[ls], mc[rs]);
    mc[cur] = Max(mc[cur], rc[ls] + lc[rs]);
    lc[cur] = Max(lc[ls], A[mid] - A[x - 1] + lc[rs]);
    rc[cur] = Max(rc[rs], A[y] - A[mid] + rc[ls]);
}
void build(int cur, int x, int y)
{
    int mid = (x + y) >> 1, ls = cur << 1, rs = cur << 1 | 1;
    if(x == y)
    {
        mc[cur] = lc[cur] = rc[cur] = a[x];
        return ;
    }
    build(ls, x, mid);
    build(rs, mid + 1, y);
    update(cur, x, y);
}
int query(int cur, int x, int y, int s, int t, int flag, int &ans)
{
    int mid = (x + y) >> 1, ls = cur << 1, rs = cur << 1 | 1;
    if(x >= s && y <= t)
    {
        ans = Max(ans, mc[cur]);
        if(flag == 0)
            return lc[cur];
        else
            return rc[cur];
    }
    if(mid >= t)
        return query(ls, x, mid, s, t, 0, ans);
    else if(mid + 1 <= s)
        return query(rs, mid + 1, y, s, t, 1, ans);
    int ln, rn;
    ln = query(ls, x, mid, s, t, 1, ans);
    rn = query(rs, mid + 1, y, s, t, 0, ans);
    ans = Max(ans, ln + rn);
    if(flag == 0)
        return Max(lc[ls], A[mid] - A[x - 1] + rn);
    else
        return Max(rc[rs], A[y] - A[mid] + ln);
}
void Search(int cur, int x, int y, int s, int t, int flag, int &ans)
{
    int mid = (x + y) >> 1, ls = cur << 1, rs = cur << 1 | 1;
    if(x >= s && y <= t)
    {
        if(flag == 0)
            ans = Max(ans, A[x - 1] - A[s - 1] + lc[cur]);
        else
            ans = Max(ans, A[t] - A[y] + rc[cur]);
        return ;
    }
    if(mid >= s)
        Search(ls, x, mid, s, t, flag, ans);
    if(mid + 1 <= t)
        Search(rs, mid + 1, y, s, t, flag, ans);
}
void init()
{
    int i, j, k;
    scanf("%d", &N);
    A[0] = 0;
    for(i = 1; i <= N; i ++)
    {
        scanf("%d", &a[i]);
        A[i] = A[i - 1] + a[i];
    }
    build(1, 1, N);
}
void solve()
{
    int i, q, ans, t1, t2, x1, y1, x2, y2;
    scanf("%d", &q);
    for(i = 0; i < q; i ++)
    {
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        if(y1 < x2)
        {
            t1 = t2 = -INF;
            Search(1, 1, N, x1, y1, 1, t1);
            Search(1, 1, N, x2, y2, 0, t2);
            printf("%d\n", t1 + t2 + A[x2 - 1] - A[y1]);
        }
        else
        {
            ans = -INF;
            query(1, 1, N, x2, y1, 0, ans);
            t1 = t2 = -INF;
            Search(1, 1, N, x1, y1, 1, t1);
            Search(1, 1, N, y1, y2, 0, t2);
            ans = Max(ans, t1 + t2 - a[y1]);
            t1 = t2 = -INF;
            Search(1, 1, N, x1, x2, 1, t1);
            Search(1, 1, N, x2, y2, 0, t2);
            ans = Max(ans, t1 + t2 - a[x2]);
            printf("%d\n", ans);
        }
    }
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t --)
    {
        init();
        solve();
    }
    return 0;
}