zoukankan      html  css  js  c++  java
  • bzoj 3653: 谈笑风生【dfs序+主席树】

    考虑b的两种情况,一种是p的祖先,这种点有min(k,de[p]-1)个,然后每个这种b都有si[p]-1个c点可选;
    另一种是p的子孙,要求是在p的子树内且deep在de[p]+1~de[p]+k之间,然后一个这样的b点贡献是si[b]-1,也就是在他的子树内选c点,考虑怎么算这个,用dfs序维护,这样每次查p的子树就是一段区间了,然后用主席树按deep维护每个点的size-1,就可以直接查了

    #include<iostream>
    #include<cstdio>
    using namespace std;
    const int N=300005;
    int n,q,h[N],cnt,de[N],si[N],l[N],r[N],con,a[N],rt[N],tot;
    struct qwe
    {
    	int ne,to;
    }e[N<<1];
    struct zxs
    {
    	int ls,rs;
    	long long s;
    }t[20000005];
    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 add(int u,int v)
    {
    	cnt++;
    	e[cnt].ne=h[u];
    	e[cnt].to=v;
    	h[u]=cnt;
    }
    void dfs(int u,int fa)
    {
    	a[++con]=u;
    	l[u]=con;
    	si[u]=1;
    	de[u]=de[fa]+1;
    	for(int i=h[u];i;i=e[i].ne)
    		if(e[i].to!=fa)
    		{
    			dfs(e[i].to,u);
    			si[u]+=si[e[i].to];
    		}
    	r[u]=con;
    }
    void ins(int &ro,int la,int l,int r,int x,int v)
    {
    	ro=++tot;
    	t[ro]=t[la];
    	t[ro].s+=v;
    	if(l==r)
    		return;
    	int mid=(l+r)>>1;
    	if(x<=mid)
    		ins(t[ro].ls,t[la].ls,l,mid,x,v);
    	else
    		ins(t[ro].rs,t[la].rs,mid+1,r,x,v);
    }
    long long ques(int ro,int la,int l,int r,int ll,int rr)
    {
    	if(ll>rr)
    		return 0;
    	if(l==ll&&r==rr)
    		return t[ro].s-t[la].s;
    	int mid=(l+r)>>1;
    	if(rr<=mid)
    		return ques(t[ro].ls,t[la].ls,l,mid,ll,rr);
    	else if(ll>mid)
    		return ques(t[ro].rs,t[la].rs,mid+1,r,ll,rr);
    	else
    		return ques(t[ro].ls,t[la].ls,l,mid,ll,mid)+ques(t[ro].rs,t[la].rs,mid+1,r,mid+1,rr);
    }
    int main()
    {
    	n=read(),q=read();
    	for(int i=1;i<n;i++)
    	{
    		int x=read(),y=read();
    		add(x,y),add(y,x);
    	}
    	dfs(1,0);
    	for(int i=1;i<=n;i++)
    		ins(rt[i],rt[i-1],1,n,de[a[i]],si[a[i]]-1);
    	while(q--)
    	{
    		int p=read(),k=read();
    		printf("%lld
    ",1ll*min(k,de[p]-1)*(si[p]-1)+ques(rt[r[p]],rt[l[p]-1],1,n,de[p]+1,min(de[p]+k,n)));
    	}
    	return 0;
    }
    
  • 相关阅读:
    Js实现继承的几种方式
    ES6-promise实现异步请求
    CSS重排和重绘
    jq在元素的后面插入内容
    yii2 Query Builder 查询打印sql语句
    js replace(a,b)替换指定字符
    linux 打印当前工作目录
    linux 查看磁盘空间
    linux查看大文件
    js-Cannot read property 'innerHTML' of null
  • 原文地址:https://www.cnblogs.com/lokiii/p/10750499.html
Copyright © 2011-2022 走看看