Codeforces Round #433 (Div. 2) E. Boredom (主席树 计算矩形内点的个数)

题意：一个n*n的网格图，有n个标记，每列只有一个标记，定义美丽的矩形为以两个标记所在位置构成的矩形（对角线的两个角）。q次询问，每次询问给你一个矩形，问有多少个美丽的矩形与该矩形相交。

思路：我们可以将整个图分成9个矩形：

1 2 3 

4 5 6

7 8 9

假设给定矩形区域是5，且第i个矩形中含有i个标记，

那么答案即为：(1*(5+6+8+9)+2*(4+5+6+7+8+9)+3*(4+5+7+8)+4*(2+3+5+6+8+9)+6*(1+2+4+5+7+8)+7*(2+3+5+6)+8*(1+2+3+4+5+6)+9*(1+2+4+5))/2+5*5/2

最关键的是怎么求矩形内有多少个标记，二维树状数组/线段树可做，但这题显然炸空间，可以用主席树来处理，按列来建立，统计时计算列之间版本的行区间的值的数量。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
int lson[maxn<<5], rson[maxn<<5], sum[maxn<<5];
int T[maxn];
int n, q, tot;

int update(int pre, int l, int r, int x)
{
    int rt = ++tot;
    lson[rt] = lson[pre], rson[rt] = rson[pre], sum[rt] = sum[pre]+1;
    if(l < r)
    {
        int mid = (l+r)/2;
        if(x <= mid)
            lson[rt] = update(lson[pre], l, mid, x);
        else
            rson[rt] = update(rson[pre], mid+1, r, x);
    }
    return rt;
}

int query(int u, int v, int l, int r, int i, int j)
{
    if(i <= l && j >= r) return sum[v]-sum[u];
    int mid = (l+r)/2;
    int ans = 0;
    if(i <= mid) ans += query(lson[u], lson[v], l, mid, i, j);
    if(j > mid) ans += query(rson[u], rson[v], mid+1, r, i, j);
    return ans;
}

int ask(int x1, int x2, int y1, int y2)
{
    if(x1 > x2 || y1 > y2) return 0;
    return query(T[x1-1], T[x2], 1, n, y1, y2);
}

int main(void)
{
    while(cin >> n >> q)
    {
        tot = 0;
        memset(sum, 0, sizeof(sum));
        T[0] = 0;
        for(int i = 1; i <= n; i++)
        {
            int x;
            scanf("%d", &x);
            T[i] = update(T[i-1], 1, n, x);
        }
        for(int i = 1; i <= q; i++)
        {
            int l, d, r, u;
            scanf("%d%d%d%d", &l, &d, &r, &u);
            ll ans = 0, tmp=ask(l, r, d, u);
            ans += (ll)ask(1, l-1, 1, d-1)*(ask(l, n, d, n)+tmp);
            ans += (ll)ask(l, r, 1, d-1)*(ask(1, n, d, n)+tmp);
            ans += (ll)ask(r+1, n, 1, d-1)*(ask(1, r, d, n)+tmp);
            ans += (ll)ask(r+1, n, d, u)*(ask(1, r, 1, n)+tmp);
            ans += (ll)ask(1, l-1, d, u)*(ask(l, n, 1, n)+tmp);
            ans += (ll)ask(1, l-1, u+1, n)*(ask(l, n, 1, u)+tmp);
            ans += (ll)ask(l, r, u+1, n)*(ask(1, n, 1, u)+tmp);
            ans += (ll)ask(r+1, n, u+1, n)*(ask(1, r, 1, u)+tmp);
            printf("%I64d\n", (ans+tmp*(tmp-1))/2);
        }
    }
    return 0;
}