NC16419. [NOIP2017]列队
描述
输入描述
输入共 q+1 行。
第 1 行包含 3 个用空格分隔的正整数 n, m, q,表示方阵大小是 n 行 m 列,一共发生了 q 次事件。
接下来 q 行按照事件发生顺序描述了 q 件事件。每一行是两个整数 x, y, 用一个空格分隔, 表示这个离队事件中离队的学生当时排在第 x 行第 y 列。
输出描述
按照事件输入的顺序,每一个事件输出一行一个整数,表示这个离队事件中离队学生的编号。
示例1
输入:
2 2 3 1 1 2 2 1 2
输出:
1 1 4
说明:
C++14(g++5.4) 解法, 执行用时: 636ms, 内存消耗: 103880K, 提交时间: 2019-08-24 19:19:26
#include<bits/stdc++.h> #define ll long long #define maxn 600010 using namespace std; ll n,m,x,y,q,N,siz,pos,rt[maxn]; vector<ll> G[maxn]; struct node{ ll sum,ls,rs; }e[maxn*40]; void modify(ll& u,ll l,ll r){ if (!u) u=++siz; e[u].sum++; if (l<r){ ll mid=l+r>>1; if (mid>=pos) modify(e[u].ls,l,mid); else modify(e[u].rs,mid+1,r); } } ll query(ll u,ll l,ll r,ll k){ if (l==r) return l; int mid=l+r>>1; int sizl=mid-l+1-e[e[u].ls].sum; if (sizl>=k) return query(e[u].ls,l,mid,k); else return query(e[u].rs,mid+1,r,k-sizl); } ll del_r(ll x,ll y){ pos=query(rt[n+1],1,N,x); modify(rt[n+1],1,N); ll ans=pos<=n?1ll*pos*m:G[n+1][pos-n-1]; G[n+1].push_back(y?y:ans); return ans; } ll del_l(ll x,ll y){ pos=query(rt[x],1,N,y); modify(rt[x],1,N); ll ans=pos<m?1ll*(x-1)*m+pos:G[x][pos-m]; G[x].push_back(del_r(x,ans)); return ans; } int main(){ scanf("%lld%lld%lld",&n,&m,&q); N=max(n,m)+q; while (q--){ scanf("%lld%lld",&x,&y); if (y==m) printf("%lld\n",del_r(x,0)); else printf("%lld\n",del_l(x,y)); } return 0; }
C++11(clang++ 3.9) 解法, 执行用时: 539ms, 内存消耗: 58984K, 提交时间: 2020-03-04 14:21:10
#include<bits/stdc++.h> using namespace std; const int NN=6000004,MM=300004; int n,m,q,rt[MM],son1[NN],son2[NN],sum[NN],tot; long long num[MM<<1]; vector<long long>g[MM]; int Query(int k,int L,int R,int &p) { if(!p) p=++tot,sum[p]=R-L+1; sum[p]--; if(L==R) return L; int mid=L+R>>1,suml=son1[p]?sum[son1[p]]:mid-L+1; return suml>=k?Query(k,L,mid,son1[p]):Query(k-suml,mid+1,R,son2[p]); } void solve() { for(int i=1;i<=n;i++) num[i]=1ll*i*m; for(int i=n+1,nn=n+q,mm=m+q,x,y,ans;q;q--,i++) { scanf("%d%d",&x,&y); if(y==m) num[i]=num[ans=Query(x,1,nn,rt[0])]; else { num[i]=(ans=Query(y,1,mm,rt[x]))<m?(x-1ll)*m+ans:g[x][ans-m]; g[x].push_back(num[Query(x,1,nn,rt[0])]); } printf("%lld\n",num[i]); } } int main() { cin>>n>>m>>q; solve(); return 0; }