zoukankan      html  css  js  c++  java
  • HNOI2015 接水果

    练码力的好题.
    题目链接


    首先容易想到一条路径代表二维平面上的一个点.
    在树上,那就求出(dfs)序.
    我们令(L_u=dfn[u],R_u=dfn[u]+size[u]-1)
    一条(u ightarrow v)的路径可以被映射到一个点((L_u,R_u))

    假设有一个盘子(u ightarrow v),我们分两种情况讨论.
    如果(lca(u,v))不为(u),且不为(v),那么只要水果一个端点在(u)的子树内,另一个端点在(v)的子树内即可.
    因此水果路径是属于({(L_u,L_v),(R_u,R_v)})这个矩形的.
    假设(lca(u,v)=u),那么找到(u)的一个儿子在(u ightarrow v)的路径上,设它为(x),那么只要一个点在(v)的子树内,一个点不在(x)的子树内即可.
    这是两个矩形:({(1,L_v),(L_x-1,R_v)})({(R_x+1,L_v),(n,R_v)}).

    那么我们把这个题转化成这样:
    有若干个矩形,每次询问一个点,求覆盖它的所有矩形中权值第(k)小的.
    我们可以整体二分.

    代码如下:
    还挺短
    好吧其实大力压行过了(QAQ)

    #include<iostream>
    #include<cstdio>
    #include<algorithm>
    #include<cstring>
    #include<cmath>
    #include<vector>
    #define N (80010)
    #define M (200010)
    #define inf (0x7f7f7f7f)
    #define rg register int
    #define Label puts("NAIVE")
    #define spa print(' ')
    #define ent print('
    ')
    #define rand() (((rand())<<(16))^(rand()))
    typedef long double ld;
    typedef long long LL;
    typedef unsigned long long ull;
    using namespace std;
    inline char read(){
    	static const int IN_LEN=1000000;
    	static char buf[IN_LEN],*s,*t;
    	return (s==t?t=(s=buf)+fread(buf,1,IN_LEN,stdin),(s==t?-1:*s++):*s++);
    }
    template<class T>
    inline void read(T &x){
    	static bool iosig;
    	static char c;
    	for(iosig=false,c=read();!isdigit(c);c=read()){
    		if(c=='-')iosig=true;
    		if(c==-1)return;
    	}
    	for(x=0;isdigit(c);c=read())x=((x+(x<<2))<<1)+(c^'0');
    	if(iosig)x=-x;
    }
    inline char readchar(){
    	static char c;
    	for(c=read();!isalpha(c)&&!isdigit(c);c=read())
    	if(c==-1)return 0;
    	return c;
    }
    const int OUT_LEN = 10000000;
    char obuf[OUT_LEN],*ooh=obuf;
    inline void print(char c) {
    	if(ooh==obuf+OUT_LEN)fwrite(obuf,1,OUT_LEN,stdout),ooh=obuf;
    	*ooh++=c;
    }
    template<class T>
    inline void print(T x){
    	static int buf[30],cnt;
    	if(x==0)print('0');
    	else{
    		if(x<0)print('-'),x=-x;
    		for(cnt=0;x;x/=10)buf[++cnt]=x%10+48;
    		while(cnt)print((char)buf[cnt--]);
    	}
    }
    inline void flush(){fwrite(obuf,1,ooh-obuf,stdout);}
    struct que{
    	int x,y,w,id;
    	friend bool operator <(que a,que b){return a.x<b.x;}
    }q[M],q1[M],q2[M];
    struct chg{
    	int ly,ry,x,w,tag,id;
    	friend bool operator <(chg a,chg b){return a.w<b.w||a.w==b.w&&a.id<b.id;}
    }a[N];
    int n,m,Q,E,fi[N],ne[M],b[M],dfn[N],ind,siz[N],f[N][17],cnt,dep[N],ans[N];
    struct BIT{
    	int c[M];
    	void add(int x,int w){for(;x<=n;x+=(x&-x))c[x]+=w;}
    	void add(int l,int r,int w){add(l,w),add(r+1,-w);}
    	int query(int x){int ans=0;for(;x;x-=(x&-x))ans+=c[x];return ans;}
    }T;
    void add(int x,int y){ne[++E]=fi[x],fi[x]=E,b[E]=y;}
    void add_line(int lx,int ly,int rx,int ry,int w){
    	if(lx>ly)swap(lx,ly),swap(rx,ry);
    	a[++cnt]=(chg){ly,ry,lx,w,1,cnt};
    	a[++cnt]=(chg){ly,ry,rx+1,w,-1,cnt};
    } 
    void dfs(int u,int pre){
    	dfn[u]=++ind,siz[u]=1,f[u][0]=pre;
    	for(int i=fi[u];i;i=ne[i]){
    		int v=b[i];if(v==pre)continue;
    		dep[v]=dep[u]+1,dfs(v,u),siz[u]+=siz[v];
    	}
    }
    void INIT(){
    	dfs(1,0);
    	for(int i=1;i<=16;i++)
    	for(int j=1;j<=n;j++)
    	f[j][i]=f[f[j][i-1]][i-1];
    }
    int lca(int x,int y){
    	if(dep[x]<dep[y])swap(x,y);
    	int k=dep[x]-dep[y];
    	for(int i=16;~i;i--)
    	if(k&(1<<i))x=f[x][i];
    	if(x==y)return x;
    	for(int i=16;~i;i--)
    	if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
    	if(x==y)return x;else return f[x][0];
    }
    int find(int x,int fa){
    	int k=dep[x]-dep[fa]-1;
    	for(int i=16;~i;i--)
    	if(k&(1<<i))x=f[x][i];
    	return x;
    }
    bool cmp(chg a,chg b){return a.x<b.x||a.x==b.x&&a.id<b.id;}
    void solve(int L,int R,int l,int r){
    	if(L>R)return;
    	if(l+1>=r){
    		for(int i=L;i<=R;i++)
    		ans[q[i].id]=a[l].w;
    		return;
    	}
    	int mid=(l+r)>>1,len=0,len1=0;
    	if(mid&1)mid++;sort(a+l,a+mid+1,cmp);
    	for(int i=l,j=L;i<=mid;i++){
    		for(;a[i].x>q[j].x&&j<=R;j++){
    			int x=T.query(q[j].y);
    			if(x>=q[j].w)q1[++len]=q[j];
    			else q2[++len1]=q[j],q2[len1].w-=x;
    		}
    		T.add(a[i].ly,a[i].ry,a[i].tag);
    		if(i==mid)for(;j<=R;j++)q2[++len1]=q[j];
    	}
    	sort(a+l,a+mid+1);
    	for(int i=1;i<=len;i++)q[L+i-1]=q1[i];
    	for(int i=1;i<=len1;i++)q[L+len+i-1]=q2[i];
    	solve(L,L+len-1,l,mid),solve(L+len,R,mid+1,r);
    }
    int main(){
    	read(n),read(m),read(Q);
    	for(int i=1,x,y;i<n;i++)
    	read(x),read(y),add(x,y),add(y,x);INIT();
    	for(int i=1,x,y,w;i<=m;i++){
    		read(x),read(y),read(w);
    		int rt=lca(x,y);
    		if(rt==x||rt==y){
    			if(rt==y)swap(x,y); int T=find(y,x);
    			if(1<=dfn[T]-1)add_line(1,dfn[y],dfn[T]-1,dfn[y]+siz[y]-1,w);
    			if(dfn[T]+siz[T]<=n)add_line(dfn[T]+siz[T],dfn[y],n,dfn[y]+siz[y]-1,w);
    		}
    		else add_line(dfn[x],dfn[y],dfn[x]+siz[x]-1,dfn[y]+siz[y]-1,w);
    	}
    	for(int i=1,x,y,K;i<=Q;i++){
    		read(x),read(y),read(K);
    		if(dfn[x]>dfn[y])swap(x,y);
    		q[i]=(que){dfn[x],dfn[y],K,i};
    	}
    	sort(a+1,a+cnt+1),sort(q+1,q+Q+1),solve(1,Q,1,cnt);
    	for(int i=1;i<=Q;i++)print(ans[i]),ent;
    	return flush(),0;
    }
    
  • 相关阅读:
    【从0安装】mysql
    Java面试题整理(待完善)
    Linux部署Java项目
    执行旧命令的几种方法
    SQL Server死锁报错分析
    枚举类中枚举值不存在.valueOf(enum) 抛异常处理
    InitializingBean的项目开发使用
    巧用枚举来干掉if-else,代码更优雅!
    使用Docker安装配置GitLab CE
    批量条件导出之---CSV
  • 原文地址:https://www.cnblogs.com/Romeolong/p/10195956.html
Copyright © 2011-2022 走看看