码迷,mamicode.com
首页 > 其他好文 > 详细

【JZOJ4811】排队

时间:2019-07-03 16:39:36      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:pre   next   analysis   get   heap   stream   com   ++i   else   

description

技术图片


analysis

  • \(+\)树上倍增

  • 考虑后序遍历搞出\(dfs\)序,那么要填肯定是从\(dfs\)序开始填

  • 把每个点是序里第几位看成优先级,用小根堆来维护当前空着的优先级最小的点

  • 插入每次弹\(x\)次堆顶,然后把这些点全部打上标记,注意标记一定是先打儿子再打父亲

  • 然后找一个点深度最浅的打过标记的祖先,由于标记肯定打完了该点到祖先的所有点,于是倍增就可以找出

  • 找完祖先后,把该祖先入堆,删除标记即可


code

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<iostream>
#define MAXN 100005
#define MAXM MAXN*2
#define ll long long
#define fo(i,a,b) for (ll i=a;i<=b;++i)
#define fd(i,a,b) for (ll i=a;i>=b;--i)

using namespace std;

vector<ll>a[MAXN];
ll b[MAXN],c[MAXN],f[MAXN],heap[MAXN],depth[MAXN];
ll pre[MAXN][17];
bool bz[MAXN];
ll n,t,tot;

inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while (ch<'0' || '9'<ch){if (ch=='-')f=-1;ch=getchar();}
    while ('0'<=ch && ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*f;
}
inline bool cmp(ll x,ll y){return f[x]<f[y];}
inline void swap(ll &x,ll &y){ll z=x;x=y,y=z;}
inline void dfs(ll x,ll y)
{
    fo(i,0,a[x].size()-1)if (a[x][i]!=y)depth[a[x][i]]=depth[x]+1,dfs(a[x][i],x);
    f[x]=++tot,pre[x][0]=y;
}
inline void insert(ll x)
{
    heap[++heap[0]]=x;ll now=heap[0];
    while (c[heap[now>>1]]>c[heap[now]] && now>1)swap(heap[now],heap[now>>1]),now>>=1;
}
inline ll get()
{
    ll tmp=heap[1],now=1;
    heap[1]=heap[heap[0]--],heap[heap[0]+1]=0;
    while ((now<<1)<=heap[0])
    {
        ll next=now<<1;
        if (next<heap[0] && c[heap[next+1]]<c[heap[next]])++next;
        if (c[heap[now]]<=c[heap[next]])break;
        swap(heap[now],heap[next]),now=next;
    }
    return tmp;
}
inline ll find(ll x)
{
    fd(i,16,0)if (bz[pre[x][i]])x=pre[x][i];
    return x;
}
int main()
{
    freopen("T1.in","r",stdin);
    n=read(),t=read();
    fo(i,1,n-1)
    {
        ll x=read(),y=read();
        a[x].push_back(y);
        a[y].push_back(x);
    }
    fo(i,1,n)sort(a[i].begin(),a[i].end()),b[i]=i;
    tot=0,dfs(1,0),sort(b+1,b+n+1,cmp);
    fo(i,1,n)c[b[i]]=i;
    fo(i,1,n)insert(b[i]);
    fo(j,1,16)fo(i,1,n)pre[i][j]=pre[pre[i][j-1]][j-1];
    while (t--)
    {
        ll op=read(),x=read(),y;
        if (op==1)
        {
            fo(i,1,x)bz[y=get()]=1;
            printf("%lld\n",y);
        }
        else
        {
            y=find(x),printf("%lld\n",depth[x]-depth[y]);
            insert(y),bz[y]=0;
        }
    }
    return 0;
}

【JZOJ4811】排队

标签:pre   next   analysis   get   heap   stream   com   ++i   else   

原文地址:https://www.cnblogs.com/horizonwd/p/11127036.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!