【HDU 1754】I Hate It —— 线段树(数组和指针实现)
I Hate It
Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 45627 Accepted Submission(s): 17904
这让很多学生很反感。
不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。
在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取‘Q‘或‘U‘) ,和两个正整数A,B。
当C为‘Q‘的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。
当C为‘U‘的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。
5 6 1 2 3 4 5 Q 1 5 U 3 6 Q 3 4 Q 4 5 U 2 9 Q 1 5
5 6 5 9HintHuge input,the C function scanf() will work better than cin
解题报告:
由于题目要求多次查询和更新数据,所以用简单的数组查询肯定会TLE。因而改用统计方法来实现,最经典的肯定是使用线段树来实现。建树,区间查询,点更新。这么简单的题一直TLE,最后发现竟然是没加 !=EOF ,一直停留在while里面,摔!!!
/* *线段树的指针实现 */ #include <iostream> #include <cstdio> using namespace std; int n,m,l,r; char c[10]; const int maxn=200010; int stu[maxn]; struct Node{ int a,b,MAX; Node* l;Node* r; Node(int x,int y):a(x),b(y),l(NULL),r(NULL){} }*root; inline int max(int a,int b){return a>b?a:b;} void build(int a,int b,Node*& rt) { if(b==a) {rt=new Node(a,b);rt->MAX=stu[a];return;} rt = new Node(a,b); int mid=(b-a+1)/2+a-1; build(a,mid,rt->l); build(mid+1,b,rt->r); rt->MAX = max(rt->l->MAX,rt->r->MAX); } int query(int a,int b,Node* rt) { if(a == rt->a && b== rt->b) return rt->MAX; int mid=rt->l->b; if(a<=mid && b>mid) return max( query(a,mid,rt->l),query(mid+1,b,rt->r) ); else{ if(b <= mid) return query(a,b,rt->l); else return query(a,b,rt->r); } } void update(const int& a,const int& score,Node* rt) { if(a == rt->a && a== rt->b) { rt->MAX = score; } else{ if(a<= rt->l->b) update(a,score,rt->l); else update(a,score,rt->r); rt->MAX = max(rt->l->MAX,rt->r->MAX); } } void destroy(Node* rt) { if(rt->l == NULL) delete rt; else { destroy(rt->l); destroy(rt->r); delete rt; } } int main() { while( scanf("%d%d",&n,&m)!=EOF ) { for(int i=1;i<=n;++i) scanf("%d",&stu[i]); build(1,n,root); for(int i=0;i<m;++i){ scanf("%s%d%d",c,&l,&r); if(c[0]=='Q') printf( "%d\n",query(l,r,root) ); else if(c[0]=='U')update(l,r,root); } destroy(root); //记得要delete掉,要不然会MLE } return 0; }
/* *线段树的数组实现 */ #include <cstdio> #include <iostream> using namespace std; struct Node{ int L,R,MAX; void set(int a,int b,int c=-1){L=a;R=b;MAX=c;} }; int n,m,a,b; char c[5]; const int maxn=200010; int stu[maxn]; Node segTree[4*maxn]; inline int max(int a,int b){return a>b?a:b;} inline int lchild(int p){return p<<1; } inline int rchild(int p){return (p<<1)|1;} void build_segment_tree(int cur,int l,int r) { if(l==r) {segTree[cur].set(l,r,stu[l]);return;} segTree[cur].set(l,r); build_segment_tree(lchild(cur), l , (l+r)>>1); build_segment_tree(rchild(cur),((l+r)>>1)+1,r); // 如果max是宏定义,不要直接把函数作为参数,否则会多次调用 segTree[cur].MAX=max(segTree[lchild(cur)].MAX,segTree[rchild(cur)].MAX); } int query(int cur,const int& l,const int& r) { if(r<segTree[cur].L || segTree[cur].R<l) return -1; if(l<=segTree[cur].L && segTree[cur].R<=r) return segTree[cur].MAX; // 如果max是宏定义,不要直接把函数作为参数,否则会多次调用 int mid=(segTree[cur].L+segTree[cur].R)>>1, ans=-1; if(l<=mid)ans=max(ans,query(lchild(cur),l,r)); if(mid<r) ans=max(ans,query(rchild(cur),l,r)); return ans; } void update(int cur,const int& s,const int& score) { if(segTree[cur].L==segTree[cur].R) {segTree[cur].MAX=score;return;} if(s<= (segTree[cur].L+segTree[cur].R)>>1 ) update(lchild(cur),s,score); else update(rchild(cur),s,score); // 如果max是宏定义,不要直接把函数作为参数,否则会多次调用 segTree[cur].MAX=max(segTree[lchild(cur)].MAX, segTree[rchild(cur)].MAX); } int main() { while( cin>>n>>m ) { for(int i=1;i<=n;++i) scanf("%d",&stu[i]); build_segment_tree(1,1,n); for(int i=0;i<m;++i){ scanf("%s%d%d",c,&a,&b); if(c[0]=='Q') {printf( "%d\n",query(1,a,b) );} else update(1,a,b); } } return 0; }
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。