P5838 [USACO19DEC]Milk Visits G(树剖+主席树

题意就是问你:给定一棵1e5大小的树,有点权,1e5次询问(u,v,w)询问从u到v的路径是否存在w的点权——很自然想到对一个序列操作的话就是主席树,只不过上了个树,1e5的复杂度支持树剖,那就剖成一条条序列暴力即可

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+15;
const int M = 1e5;
#define pb push_back
int n,m;
vector<int>E[N];
int dep[N],fa[N],siz[N],son[N];
void dfs1(int u,int f){
	siz[u]=1;
	for(auto v:E[u])if(v!=f){
		dep[v]=dep[u]+1;
		fa[v]=u;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[son[u]]<siz[v])
			son[u]=v;
	}
}
int rev[N],dfn[N],top[N],tot;
void dfs2(int u,int t){
	top[u]=t;
	rev[++tot]=u;
	dfn[u]=tot;
	if(!son[u]) return;
	dfs2(son[u],t);
	for(auto v:E[u]){
		if(v!=fa[u] && v!=son[u])
			dfs2(v,v);
	}
}

int tr[200005],lc[200005*20],rc[200005*20],sm[200005*20],cnt;
void update(int &rt,int pre,int l,int r,int pos){
	rt=++cnt;
	lc[rt]=lc[pre];
	rc[rt]=rc[pre];
	sm[rt]=sm[pre]+1;
	if(l==r){
		return;
	}
	int mid=l+r>>1;
	if(pos<=mid)
		update(lc[rt],lc[pre],l,mid,pos);
	else
		update(rc[rt],rc[pre],mid+1,r,pos);
}	
int query(int now,int pre,int l,int r,int pos){
	if(l==r)
		return sm[now]-sm[pre];
	int mid=l+r>>1;
	int ans=0;
	if(pos<=mid)
		ans+=query(lc[now],lc[pre],l,mid,pos);
	else
		ans+=query(rc[now],rc[pre],mid+1,r,pos);
	return ans;
}
int query(int u,int v,int w){
	int ans=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]])
			swap(u,v);
		ans+=query(tr[dfn[top[u]]-1],tr[dfn[u]],1,M,w);
		u=fa[top[u]];
	}
	if(dep[u]<dep[v])
		swap(u,v);
	ans+=query(tr[dfn[v]-1],tr[dfn[u]],1,M,w);
	return ans;
}
int w[N];
int main(){
	ios::sync_with_stdio(false);cin.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>w[i];
	for(int i=1;i<n;i++){
		int u,v;cin>>u>>v;
		E[u].pb(v);E[v].pb(u);
	}
	dfs1(1,1);
	dfs2(1,1);
	for(int i=1;i<=n;i++){
		update(tr[i],tr[i-1],1,M,w[rev[i]]);
	}
	while(m--){
		int u,v,w;cin>>u>>v>>w;
		if(query(u,v,w))
			cout<<1;
		else
			cout<<0;
	}
	cout<<'\n';
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是P4087 [USACO17DEC]Milk Measurement的c++代码: ```c++ #include<bits/stdc++.h> using namespace std; int n,d,i,x,minn=1e9,maxn=-1e9,sum=7;//注意sum要初始化为7,因为一开始有三个人挤奶! map<int,int> mp; struct node{ int day,milk,id;//day表示某一天,milk表示这一天的产奶量,id表示这头牛的编号 }a[100010]; bool cmp(node x,node y){ return x.day<y.day; } int main(){ scanf("%d%d",&n,&d); for(i=1;i<=n;i++){ scanf("%d%d%d",&a[i].day,&a[i].id,&a[i].milk); minn=min(minn,a[i].id);//记录最小的牛的编号 maxn=max(maxn,a[i].id);//记录最大的牛的编号 } sort(a+1,a+n+1,cmp);//排序 for(i=1;i<=n;i++){ int p=a[i].id; mp[p]+=a[i].milk;//记录每头牛产奶总量 if(mp[p]-a[i].milk>=mp[minn]&&mp[p]>=mp[minn]){//如果这头牛的产奶总量减去这一天的产奶量后等于最小产奶量且这头牛的产奶总量大于等于最小产奶量 sum--; } if(mp[p]>=mp[maxn]&&mp[p]-a[i].milk<mp[maxn]){//如果这头牛的产奶总量大于等于最大产奶量且这头牛的产奶总量减去这一天的产奶量小于最大产奶量 sum++; } if(mp[p]-a[i].milk<mp[maxn]&&mp[p]>=mp[maxn]){//如果这头牛的产奶总量减去这一天的产奶量小于最大产奶量且这头牛的产奶总量大于等于最大产奶量 if(mp[maxn]-mp[p]+a[i].milk>0)sum++; } mp[p]-=a[i].milk;//减去这一天的产奶量 if(i==n||a[i].day!=a[i+1].day){//如果到了新的一天或者到了最后一天 if(mp[maxn]!=mp[a[i].id]&&mp[a[i].id]>=mp[maxn])sum++;//如果这头牛的产奶总量不等于最大产奶量且这头牛的产奶总量大于等于最大产奶量 if(mp[maxn]==mp[a[i].id]){//如果这头牛的产奶总量等于最大产奶量 if(a[i].id==maxn)sum+=0;//如果这头牛就是最大产奶量的牛,那么不需要增加计数器 else sum++;//否则需要增加计数器 } if(mp[minn]!=mp[a[i].id]&&mp[a[i].id]>=mp[minn])sum++;//如果这头牛的产奶总量不等于最小产奶量且这头牛的产奶总量大于等于最小产奶量 if(mp[minn]==mp[a[i].id]){ if(a[i].id==minn)sum+=0;//如果这头牛就是最小产奶量的牛,那么不需要增加计数器 else sum++;//否则需要增加计数器 } } } printf("%d\n",sum); return 0; } ``` 该题的解题思路是模拟,需要注意细节问题。我们可以首先将输入的数据按天数排序,然后模拟每一天挤奶的情况,并根据题目要求进行计数即可。具体细节请见代码注释。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值