题意:输入一个t,表示有t组测试数据;
接下来一行,输入两个数,k,m,其中k表示这个辆车最多可以坐这么多人,m表示有m次询问能否上车;
每一次询问,输入两个数a,b,表示该乘客能否在a站台上车,b站台下车,乘车区间为(a,b--),先后次序;
即我每次询问,你就判断在a站台处将会有多少人还在车上,小于k则表示能够上车,更新数据,反之不能上车;
解题思路:这道题很明显就是线段树的区间更新,即判断线段的重叠次数。
按照刘汝佳书上写的代码WA了,不知道为什么。。。
#include<iostream> #include<cstdio> #include<cstring> using namespace std; const int maxn = 1e6+5; int add[maxn<<2],Max[maxn<<2],l,r,_max; int ans[maxn]; void maintain(int o,int L,int R) { int lc = o*2,rc = o*2+1; Max[o] = 0; if(R > L){ Max[o] = max(Max[lc],Max[rc]); } Max[o] += add[o]; } void update(int o,int L,int R,int v) { int lc = o*2, rc = o*2+1; if(l <= L && R <= r){ add[o] += v; } else{ int M = (L + R) >> 1; if(M >= l) update(lc,L,M,v); if(r > M) update(rc,M+1,R,v); } maintain(o,L,R); } void query(int o,int L,int R,int addv) { int lc = o*2, rc = o*2+1; if(l <= L && R <= r){ _max = max(_max,Max[o] + addv); } else{ int M = (L + R) >> 1; if(l <= M) query(lc,L,M,addv+add[lc]); if(r > M) query(rc,M+1,R,addv+add[rc]); } } int main() { int t,cas = 1,len; scanf("%d",&t); while(t--){ memset(add,0,sizeof(add)); memset(Max,0,sizeof(Max)); len = 0; int k,q; scanf("%d%d",&k,&q); for(int i = 1; i <= q; i++){ scanf("%d%d",&l,&r); r--; _max = 0; query(1,1,1000000,0); if(_max < k){ update(1,1,1000000,1); ans[len++] = i; } } printf("Case %d:\n",cas++); for(int i = 0; i < len; i++) printf("%d ",ans[i]); printf("\n\n"); } return 0; }
看了别人的代码,采用了一种lazy思想,也就是用一层更新一层,如果我在某一层能够找到符合要求的区间,那么就不更新它的子节点了,等到我需要它的子节点时,再把子节点更新了。。。总之是这样的一种思想,结合代码多思考下。。。
#include<iostream> #include<string> #include<algorithm> #include<cstdio> #include<cstring> using namespace std; const int N=1000005; int ans[N]; struct node { int l,r,v,lazy; }node[N<<2]; // 线段树的空间大概是数组空间的4倍; void build(int l,int r,int numb) // 线段树的建立; { node[numb].l=l; node[numb].r=r; node[numb].v=0; node[numb].lazy=0; // 用了lazy思想,提高了效率; if(l==r) return; int mid=(l+r)>>1; build(l,mid,numb<<1); build(mid+1,r,numb<<1|1); } void PushUp(int numb) // 往上往父节点方向更新数据;但是这里不是左右儿子的和,而是最大值,因为是站台人数; { node[numb].v=max(node[numb<<1].v,node[numb<<1|1].v); } void PushDown(int numb) // 向下往左右儿子方向更新数据; { node[numb<<1].lazy+=node[numb].lazy; node[numb<<1|1].lazy+=node[numb].lazy; node[numb<<1].v+=node[numb].lazy; node[numb<<1|1].v+=node[numb].lazy; node[numb].lazy=0; // 更新完了要清零; } void Insert(int l,int r,int numb) // 插入更新数据; { if(node[numb].l>=l&&node[numb].r<=r) // 如果区间完全重合,则不需要再往下更新了,先保存起来,可以节约很多的时间(lazy思想) { node[numb].v+=1; node[numb].lazy+=1; return; } if(node[numb].lazy) PushDown(numb); // 因为没有找到完全重合的区间,所以要先更新下一层区间; int mid=(node[numb].r+node[numb].l)>>1; if(l>mid) Insert(l,r,numb<<1|1); else if(r<=mid) Insert(l,r,numb<<1); else{ Insert(l,mid,numb<<1); Insert(mid+1,r,numb<<1|1); } PushUp(numb); // 最后还得往上返回,更新父节点区间; } int query(int l,int r,int numb) // 查询区间l到r; { if(node[numb].l>=l&&node[numb].r<=r){ return node[numb].v; } if(node[numb].lazy) PushDown(numb); // 道理同48行; int mid=(node[numb].r+node[numb].l)>>1; if(l>mid) return query(l,r,numb<<1|1); else if(r<=mid) return query(l,r,numb<<1); else{ return max(query(l,mid,numb<<1),query(mid+1,r,numb<<1|1)); // 道理同28行; } } int main() { int t,Case=1,len=0,k,m,a,b; scanf("%d",&t); while(t--){ len=0; memset(ans,0,sizeof(ans)); scanf("%d%d",&k,&m); build(1,1000000,1); for(int i=0;i<m;i++){ scanf("%d%d",&a,&b); b--; // 这里有一个问题,就是乘客从a上车,b下车,所以乘客在车上的区间为(a,b--); if(query(a,b,1)<k){ // 表示可以上车; ans[len++]=i+1; Insert(a,b,1); } } printf("Case %d:\n",Case++); for(int i=0; i<len; i++) printf("%d ",ans[i]); printf("\n\n"); } return 0; }