BZOJ 1537: [POI2005]Aut- The Bus 树状数组

Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 390 Solved: 266

Description

Byte City 的街道形成了一个标准的棋盘网络 – 他们要么是北南走向要么就是西东走向. 北南走向的路口从 1 到 n编号, 西东走向的路从1 到 m编号. 每个路口用两个数(i, j) 表示(1 <= i <= n, 1 <= j <= m). Byte City里有一条公交线, 在某一些路口设置了公交站点. 公交车从 (1, 1) 发车, 在(n, m)结束.公交车只能往北或往东走. 现在有一些乘客在某些站点等车. 公交车司机希望在路线中能接到尽量多的乘客.帮他想想怎么才能接到最多的乘客.

Input

第一行三个数n, m 和 k – 表示北南走向的路的个数以及西东走向的路和乘客等车的站点的个数. ( 1 <= n <= 10^9, 1 <= m <= 10^9, 1 <= k <= 10^5). 接下来k 行每行描述一个公交站的信息.第 i + 1 行三个正整数 xi, yi 和 pi, 1 <= xi <= n, 1 <= yi <= m, 1 <= pi <= 10^6. 表示在(xi, yi) 有 pi 个乘客在等车. 每个路口在数据中最多出现一次,乘客总数不会超过1 000 000 000.

Output

一个数表示最多能接到的乘客数量.

Sample Input

8 7 11

4 3 4

6 2 4

2 3 2

5 6 1

2 5 2

1 5 5

2 1 1

3 1 1

7 7 1

7 4 2

8 6 2

Sample Output

11

HINT

Source

---

有些时候我们只需要两个数字之间的大小关系，于是我们选择离散化，好久没写需要离散化的题目了，之前写主席树的时候是用stl做的离散化，没想到还有O（n）的操作，233我好菜啊，这题思路显然，就是找一个最长链满足最大就可以了，最开始以为在每个点的时候直接贪心往下选取所有可以选择的点就可以了，然而其实并不是这样，因为比如有一些点，他们的x坐标是相同的，这个时候你会去选择y坐标大的那个，然而此时不一定优，于是我们用树状数组维护max值而不是和即可，这个地方是个坑点呢（当然也有可能是我太菜了233）

---

#include <bits/stdc++.h>
using namespace std;
struct Re{ int y, rnk; } re[111111];
bool cmp1( const Re &a, const Re &b ) { return a.y < b.y; }
struct Po{ int x, y, cnt; }pos[100010];
bool cmp2( const Po &a, const Po &b ) {
    if( a.x != b.x ) return a.x < b.x;
    else             return a.y < b.y;
}
int C[100010], n, m, k, poi, ans;
inline int lowbit( int x ) { return x & ( -x ); }
void modify( int x, int val ) {
    while( x <= poi ) {
        C[x] = max( C[x], val );
        x += lowbit(x);
    }
}
int query( int x ) {
    int tmp = 0;
    while( x ) {
        tmp = max( tmp, C[x] );
        x -= lowbit(x);
    }
    return tmp;
}
int main( ) {
    scanf( "%d%d%d", &n, &m, &k );
    for( register int i = 1; i <= k; i++ ) { scanf( "%d%d%d", &pos[i].x, &re[i].y, &pos[i].cnt ); re[i].rnk = i; }
    sort( re + 1, re + k + 1, cmp1 );
    re[0].y = -1;
    for( register int i = 1; i <= k; i++ ) {
        if( re[i].y != re[ i - 1 ].y ) poi++;
        pos[re[i].rnk].y = poi;
    }
    sort( pos + 1, pos + k + 1, cmp2 ); int tmp;
    for( register int i = 1; i <= k; i++ ) {
        tmp = pos[i].cnt + query( pos[i].y );
        modify( pos[i].y, tmp );
        ans = max( ans, tmp );
    }
    printf( "%d\n", ans );
}