[hdu-6638]Snowy Smile 线段树维护 带修改的区间最大字段和 最大子矩阵 2019多校6

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define lcnow now<<1
 4 #define rcnow (now<<1)|1
 5 int const maxn=2010;
 6 long long  n,sx[maxn],sy[maxn],mx,my,ans;
 7 struct tree{
 8     long long  lm,rm,sm,tm;
 9 }tr[maxn<<4];
10 struct node{
11     long long  a, b,v;
12 }no[maxn];
13 inline long long  get_num(){
14     char ch;
15     bool flag=false;
16     long long  num=0;
17     ch=getchar();
18     while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)flag=true;ch=getchar();}
19     while(ch>=‘0‘&&ch<=‘9‘){num=(num<<3)+(num<<1)+ch-‘0‘;ch=getchar();}
20     if(flag)return -1*num;
21     return num;
22 }
23 bool cmp2(node x,node y){
24     return x.a<y.a;
25 }
26 void gethash(){
27     sort(sx+1,sx+1+n);sort(sy+1,sy+1+n);
28         mx=unique(sx+1,sx+1+n)-sx-1;
29         my=unique(sy+1,sy+1+n)-sy-1;
30         long long ax,ay;
31         for(int i=1;i<=n;i++){
32             ax=lower_bound(sx+1,sx+1+mx,no[i].a)-sx;
33             ay=lower_bound(sy+1,sy+1+my,no[i].b)-sy;
34             no[i].a=ax,no[i].b=ay;
35         }
36     sort(no+1,no+1+n,cmp2);
37 }
38 void update(int now){//利用儿子更新父亲 
39     tr[now].tm=tr[lcnow].tm+tr[rcnow].tm;
40     tr[now].lm=max(tr[lcnow].lm,tr[lcnow].tm+tr[rcnow].lm);
41     tr[now].rm=max(tr[rcnow].rm,tr[rcnow].tm+tr[lcnow].rm);
42     tr[now].sm=max(tr[lcnow].sm,tr[rcnow].sm);
43     tr[now].sm=max(tr[lcnow].rm+tr[rcnow].lm,tr[now].sm);
44 }
45 void build(int now,int l,int r){
46     if(l==r){
47         tr[now].lm=tr[now].rm=tr[now].sm=tr[now].tm=0;
48         return;
49     }
50     int mid=(l+r)>>1;
51       build(now<<1,l,mid);
52       build((now<<1)+1,mid+1,r);
53       update(now);
54 } //初始化 
55 
56 void change(int now,int l,int r,int x,int v){
57     //当前点编号now，当前点控制区间 [l,r],要修改的点编号x，加上v 
58     if(l==r){
59         tr[now].tm=tr[now].lm=tr[now].rm=tr[now].sm=tr[now].sm+v;
60         return;
61     }
62     int mid=(l+r)>>1;
63     if(x<=mid)change(now<<1,l,mid,x,v);
64     else change((now<<1)+1,mid+1,r,x,v);
65     update(now);
66 } 
67 void work(){
68     memset(tr,0,sizeof tr);
69     ans=0;
70     int k;
71     for(int i=1;i<=n;i++)if(i==1||no[i].a!=no[i-1].a){
72         build(1,1,my);
73         for(int j=i;j<=n;j=k){
74             for( k=j;k<=n&&no[k].a==no[j].a;k++){
75                 change(1,1,my,no[k].b,no[k].v);
76             }
77             if(tr[1].sm>ans)ans=tr[1].sm;
78         }
79     }
80 }
81 int main(){
82     long long t;
83     t=get_num();
84     while(t--){
85         n=get_num();
86         for(int i=1;i<=n;i++){
87             no[i].a=get_num(),no[i].b=get_num();no[i].v=get_num();
88             sx[i]=no[i].a,sy[i]=no[i].b;
89         }
90         gethash();
91         work();
92 
93         printf("%lld\n",ans);
94     }
95     return 0;
96 }