洛谷 P3960 列队【线段树】
生活随笔
收集整理的這篇文章主要介紹了
洛谷 P3960 列队【线段树】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
用動態開點線段樹分別維護每一行和最后一列,線段樹的作用是記錄被選的點的個數以及查詢第k個沒被選的點,每次修改,從行里標記被選的點,從最后一列標記向左看齊之后少的點,然后用vector維護行列的新增點
#include<iostream> #include<cstdio> #include<vector> using namespace std; const int N=600005; int n,m,q,mx,tot,rt[N]; vector<long long>a[N]; struct qwe {int ls,rs,s; }t[10000005]; int read() {int r=0,f=1;char p=getchar();while(p>'9'||p<'0'){if(p=='-')f=-1;p=getchar();}while(p>='0'&&p<='9'){r=r*10+p-48;p=getchar();}return r*f; } void update(int &ro,int l,int r,int p) {if(!ro)ro=++tot;t[ro].s++;if(l==r)return;int mid=(l+r)>>1;if(p<=mid)update(t[ro].ls,l,mid,p);elseupdate(t[ro].rs,mid+1,r,p); } int ques(int ro,int l,int r,int k) {if(l==r)return l;int mid=(l+r)>>1;if(mid-l+1-t[t[ro].ls].s>=k)return ques(t[ro].ls,l,mid,k);elsereturn ques(t[ro].rs,mid+1,r,k-(mid-l+1-t[t[ro].ls].s)); } int main() {n=read(),m=read(),q=read(),mx=max(m,n)+q;while(q--){int x=read(),y=read();if(y==m){int p=ques(rt[n+1],1,mx,x);update(rt[n+1],1,mx,p);long long ans=p<=n?1ll*p*m:a[n+1][p-n-1];a[n+1].push_back(ans);printf("%lld\n",ans);}else{int p=ques(rt[x],1,mx,y);update(rt[x],1,mx,p);;long long ans=p<m?1ll*(x-1)*m+p:a[x][p-m];p=ques(rt[n+1],1,mx,x);update(rt[n+1],1,mx,p);long long ans2=p<=n?1ll*p*m:a[n+1][p-n-1];a[n+1].push_back(ans);a[x].push_back(ans2);printf("%lld\n",ans);}}return 0; }轉載于:https://www.cnblogs.com/lokiii/p/9709749.html
總結
以上是生活随笔為你收集整理的洛谷 P3960 列队【线段树】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 咖啡文化:各种咖啡名称与选择
- 下一篇: BZOJ3574 HNOI2014抄卡组