首页 > 代码库 > UVA 1513 Movie collection (树状数组+反向存储)
UVA 1513 Movie collection (树状数组+反向存储)
题意:给你n盘歌碟按照(1....n)从上到下放,接着m个询问,每一次拿出x碟,输出x上方有多少碟并将此碟放到开头
直接想其实就是一线段的区间更新,单点求值,但是根据题意我们可以这样想
首先我们倒着存 n--1,接着每次询问时把放碟子放到最后,这样我们要开一个映射数组映射每个碟子在哪个位置
其中我们需要使用树状数组维护每个绝对位置是否有碟子(有些碟子已经放到了后面了),再使用区间求和就好了
#include<set>#include<map>#include<queue>#include<stack>#include<cmath>#include<vector>#include<string>#include<cstdio>#include<cstring>#include<stdlib.h>#include<iostream>#include<algorithm>using namespace std;#define eps 1E-8/*注意可能会有输出-0.000*/#define Sgn(x) (x<-eps? -1 :x<eps? 0:1)//x为两个浮点数差的比较,注意返回整型#define Cvs(x) (x > 0.0 ? x+eps : x-eps)//浮点数转化#define zero(x) (((x)>0?(x):-(x))<eps)//判断是否等于0#define mul(a,b) (a<<b)#define dir(a,b) (a>>b)typedef long long ll;typedef unsigned long long ull;const int Inf=1<<28;const double Pi=acos(-1.0);const int Mod=1e9+7;const int Max=200010;//开成两倍,因为要向后添加int mp[Max],bit[Max],tot,tol;//映射位置 被映射的树状数组 总大小int lowbit(int x){ return x&(-x);}void Add(int x,int y){ while(x<=tol) { bit[x]+=y; x+=lowbit(x); } return ;}int Sum(int x){ int sum=0; while(x) { sum+=bit[x]; x-=lowbit(x); } return sum;}void Init(int n,int m){ memset(bit,0,sizeof(bit)); tot=n; tol=n+m; for(int i=1; i<=n; ++i)//倒序存储 { mp[i]=n-i+1; Add(mp[i],1); } return;}int Solve(int n,int m){ ll ans=0; int pos=mp[m]; //printf("\npos=%d\n",pos); ans=n-Sum(pos);//求后面的 tot++; Add(pos,-1);//更新 Add(tot,1); mp[m]=tot;//映射到开头 return ans;}int main(){ int t,n,m,num; scanf("%d",&t); while(t--) { scanf("%d %d",&n,&m); Init(n,m); for(int i=0;i<m;++i) { scanf("%d",&num); printf("%d%c",Solve(n,num),i==m-1?‘\n‘:‘ ‘); } } return 0;}
UVA 1513 Movie collection (树状数组+反向存储)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。