链接:登录—专业IT笔试面试备考平台_牛客网 来源:牛客网 ?
题目描述
小A这次来到一个景区去旅游,景区里面有N个景点,景点之间有N-1条路径。小A从当前的一个景点移动到下一个景点需要消耗一点的体力值。但是景区里面有两个景点比较特殊,它们之间是可以直接坐观光缆车通过,不需要消耗体力值。而小A不想走太多的路,所以他希望你能够告诉它,从当前的位置出发到他想要去的那个地方,他最少要消耗的体力值是多少。
输入描述:
第一行一个整数N代表景区的个数。
接下来N-1行每行两个整数u,v代表从位置u到v之间有一条路径可以互相到达。
接下来的一行两个整数U,V表示这两个城市之间可以直接坐缆车到达。
接下来一行一个整数Q,表示有Q次询问。
接下来的Q行每行两个整数x,y,代表小A的位置在x,而他想要去的地方是y。
输出描述:
对于每个询问下x,y输出一个结果,代表x到y消耗的最少体力对于每个询问下x,y输出一个结果,代表x到y消耗的最少体力对于每个询问下x,y输出一个结果,代表x到y消耗的最少体力
输入
复制
4
1 2
1 3
2 4
3 4
2
1 3
3 4
输出
复制
1
0
题解
#include<iostream> #include<queue> #include<cstring> #include<cstdio> using namespace std; #define int long long const int maxn=3e5+5; const int INF=1e9+5; int head[maxn<<1];int d[maxn]; int fa[maxn][20]; struct node{ ? ? int to,next; }edge[maxn<<1]; int cnt1=0; void add_edge(int from,int to) { ? ? cnt1++; ? ? edge[cnt1].to=to; ? ? edge[cnt1].next=head[from]; ? ? head[from]=cnt1; } int lca(int a,int b) { ? ? if(d[a]<d[b]) swap(a,b); ? ? for(int i=19;i>=0;i--) ? ? { ? ? ? ? if(d[fa[a][i]]>=d[b]) a=fa[a][i]; ? ? } ? ? if(a==b) return a; ? ? for(int i=19;i>=0;i--) ? ? { ? ? ? ? if(fa[a][i]!=fa[b][i]) ? ? ? ? { ? ? ? ? ? ? a=fa[a][i]; ? ? ? ? ? ? b=fa[b][i]; ? ? ? ? } ? ? } ? ? return fa[a][0]; } queue<int> qu; signed main() { ? ? int n;scanf("%lld",&n); ? ? memset(head,-1,sizeof(head)); ? ? memset(fa,-1,sizeof(fa)); ? ? for(int i=1;i<=n;i++) d[i]=INF; ? ? for(int i=0;i<n-1;i++) ? ? { ? ? ? ? int from,to; ? ? ? ? scanf("%lld%lld",&from,&to); ? ? ? ? add_edge(from,to); ? ? ? ? add_edge(to,from); ? ? } ? ? int a,b;scanf("%lld%lld",&a,&b); ? ? d[1]=0;qu.push(1); ? ? while(!qu.empty()) ? ? { ? ? ? ? int from=qu.front();qu.pop(); ? ? ? ? for(int i=head[from];i!=-1;i=edge[i].next) ? ? ? ? { ? ? ? ? ? ? int to=edge[i].to; ? ? ? ? ? ? if(d[to]>d[from]+1) ? ? ? ? ? ? { ? ? ? ? ? ? ? ? d[to]=d[from]+1; ? ? ? ? ? ? ? ? fa[to][0]=from; ? ? ? ? ? ? ? ? qu.push(to); ? ? ? ? ? ? } ? ? ? ? } ? ? } ? ? for(int k=1;k<20;k++) ? ? { ? ? ? ? for(int j=1;j<=n;j++) ? ? ? ? { ? ? ? ? ? ? if(fa[j][k-1]!=-1) ? ? ? ? ? ? { ? ? ? ? ? ? ? ? fa[j][k]=fa[fa[j][k-1]][k-1]; ? ? ? ? ? ? } ? ? ? ? } ? ? } ? ? int q;scanf("%lld",&q); ? ? while(q--) ? ? { ? ? ? ? int from,to;scanf("%lld%lld",&from,&to); ? ? ? ? int v=lca(from,to); ? ? ? ? int ans=INF; ? ? ? ? ans=min(ans,d[from]+d[to]-2*d[v]); ? ? ? ? int b1=lca(from,b);int a1=lca(a,to); ? ? ? ? ans=min(ans,d[from]+d[b]-2*d[b1]+d[to]+d[a]-2*d[a1]); ? ? ? ? a1=lca(from,a),b1=lca(b,to); ? ? ? ? ans=min(ans,d[from]+d[a]-2*d[a1]+d[to]+d[b]-2*d[b1]); ? ? ? ? printf("%lld\n",ans); ? ? } }
链接:登录—专业IT笔试面试备考平台_牛客网 来源:牛客网 ?
题目描述
已知有?nnn 个节点,有?n?1n-1n?1 条边,形成一个树的结构。
给定一个根节点 kkk,每个节点都有一个权值,节点i的权值为 viv_ivi?。
给?mmm 个操作,操作有两种类型:
1 a x :表示将节点?aaa 的权值加上?xxx
2 a :表示求?aaa 节点的子树上所有节点的和(包括?aaa 节点本身)
输入描述:
第一行给出三个正整数 n,m,kn,m,kn,m,k,表示树的节点数、操作次数、和这棵树的根节点.
第二行给出?nnn 个正整数,第?iii 个正整数表示第 iii?个节点的权值?valival_ivali?
下面 n?1n-1n?1?行每行两个正整数 u,vu,vu,v,表示边的两个端点
接下来?mmm 行,每行给出一个操作
输出描述:
对于每个类型为 2 的操作,输出一行一个正整数,表示以?aaa 为根的子树的所有节点的权值和
示例1
输入
复制5 6 1 1 2 3 4 5 1 3 1 2 2 4 2 5 1 2 10 1 3 10 1 4 5 1 5 1 2 3 2 2
5 6 1
1 2 3 4 5
1 3
1 2
2 4
2 5
1 2 10
1 3 10
1 4 5
1 5 1
2 3
2 2
输出
复制13 27
13
27
链接:https://ac.nowcoder.com/acm/contest/27836/C 来源:牛客网 ?
题目描述
已知有?nnn 个节点,有?n?1n-1n?1 条边,形成一个树的结构。
给定一个根节点 kkk,每个节点都有一个权值,节点i的权值为 viv_ivi?。
给?mmm 个操作,操作有两种类型:
1 a x :表示将节点?aaa 的权值加上?xxx
2 a :表示求?aaa 节点的子树上所有节点的和(包括?aaa 节点本身)
输入描述:
第一行给出三个正整数 n,m,kn,m,kn,m,k,表示树的节点数、操作次数、和这棵树的根节点.
第二行给出?nnn 个正整数,第?iii 个正整数表示第 iii?个节点的权值?valival_ivali?
下面 n?1n-1n?1?行每行两个正整数 u,vu,vu,v,表示边的两个端点
接下来?mmm 行,每行给出一个操作
输出描述:
对于每个类型为 2 的操作,输出一行一个正整数,表示以?aaa 为根的子树的所有节点的权值和
示例1
输入
复制5 6 1 1 2 3 4 5 1 3 1 2 2 4 2 5 1 2 10 1 3 10 1 4 5 1 5 1 2 3 2 2
5 6 1
1 2 3 4 5
1 3
1 2
2 4
2 5
1 2 10
1 3 10
1 4 5
1 5 1
2 3
2 2
输出
复制13 27
13
27
#include<iostream> #include<cstring> #include<cstdio> using namespace std; #define int long long const int maxn=1e6+5; int n; int past[maxn],now[maxn],val[maxn],top[maxn],depth[maxn],fa[maxn],son[maxn],s[maxn],head[maxn<<1],ctv[maxn]; struct node1{ ? ? int to,next; }edge[maxn<<1]; int cc; void add_edge(int from,int to) { ? ? cc++; ? ? edge[cc].to=to; ? ? edge[cc].next=head[from]; ? ? head[from]=cc; } struct node{ ? ? int sum; }tree[maxn*4]; void Pushup(int rt) { ? ? tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum; } void build(int rt,int l,int r) { ? ? if(l==r){ ? ? ? ? tree[rt].sum=val[past[l]]; ? ? ? ? return ; ? ? } ? ? int m=(l+r)/2; ? ? build(rt<<1,l,m); ? ? build(rt<<1|1,m+1,r); ? ? Pushup(rt); } void change(int rt,int l,int r,int pos,int c) { ? ? if(l==r){ ? ? ? ? tree[rt].sum+=c; ? ? ? ? return ; ? ? } ? ? int m=(l+r)/2; ? ? if(pos<=m) change(rt<<1,l,m,pos,c); ? ? if(pos>m) change(rt<<1|1,m+1,r,pos,c); ? ? Pushup(rt); } int query(int rt,int l,int r,int L,int R) { ? ? int ans=0; ? ? if(l>=L&&r<=R) ? ? { ? ? ? ? return tree[rt].sum; ? ? } ? ? int m=(l+r)/2; ? ? if(L<=m) ans+=query(rt<<1,l,m,L,R); ? ? if(R>m) ans+=query(rt<<1|1,m+1,r,L,R); ? ? return ans; } void init() { ? ? memset(head,-1,sizeof(head)); } void dfs1(int root) { ? ? depth[root]=depth[fa[root]]+1; ? ? s[root]=1; ? ? for(int i=head[root];i!=-1;i=edge[i].next) ? ? { ? ? ? ? int to=edge[i].to; ? ? ? ? if(to!=fa[root]) ? ? ? ? { ? ? ? ? ? ? fa[to]=root; ? ? ? ? ? ? dfs1(to); ? ? ? ? ? ? s[root]+=s[to]; ? ? ? ? ? ? if(s[son[root]]<s[to]) ? ? ? ? ? ? { ? ? ? ? ? ? ? ? son[root]=to; ? ? ? ? ? ? } ? ? ? ? } ? ? } } int id=0; void dfs2(int root,int f) { ? ? top[root]=f; ? ? now[root]=++id; ? ? past[id]=root; ? ? if(son[root]) dfs2(son[root],f); ? ? for(int i=head[root];i!=-1;i=edge[i].next) ? ? { ? ? ? ? int to=edge[i].to; ? ? ? ? if(to!=fa[root]&&to!=son[root]) ? ? ? ? { ? ? ? ? ? ? dfs2(to,to); ? ? ? ? } ? ? } ? ? ctv[root]=id; } signed main() { ? ? int m;int k; ? ? scanf("%lld%lld%lld",&n,&m,&k); ? ? init(); ? ? for(int i=1;i<=n;i++) scanf("%lld",&val[i]); ? ? for(int i=0;i<n-1;i++) ? ? { ? ? ? ? int from,to; scanf("%lld%lld",&from,&to); ? ? ? ? add_edge(from,to); ? ? ? ? add_edge(to,from); ? ? } ? ? dfs1(k);dfs2(k,k); ? ? build(1,1,n); ? ? while(m--) ? ? { ? ? ? ? int op;scanf("%lld",&op); ? ? ? ? if(op==1) ? ? ? ? { ? ? ? ? ? ? int a,x;scanf("%lld%lld",&a,&x); ? ? ? ? ? ? change(1,1,n,now[a],x); ? ? ? ? } ? ? ? ? if(op==2) ? ? ? ? { ? ? ? ? ? ? int ?a;scanf("%lld",&a); ? ? ? ? ? ?printf("%lld\n",query(1,1,n,now[a],ctv[a])); ? ? ? ? } ? ? } }
|