Tree
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 524288/524288 K (Java/Others)
Total Submission(s): 185 Accepted Submission(s): 64
Problem Description
wls 有三棵树,树上每个节点都有一个值 ai,现在有 2 种操作:
- 将一条链上的所有节点的值开根号向下取整;
- 求一条链上值的和;
链的定义是两点之间的最短路。
Input
第一行两个数 n, q 分别代表树上点的数量和操作数量。
第二行 n 个整数,第 i 个数代表第 i 个点的值 ai。
接下来 n − 1 行, 每行两个整数 u, v 代表 u,v 之间有一条边。数据保证点两两联通。
接下来 q 行,每行有个整数 op, u, v,op = 0 表示将 u, v 这条链上所有的点的值开根号向下取整,op = 1表示询问 u,v 这条链上的值的和。
1 ≤ n, q ≤ 100, 000
0 ≤ ai ≤ 1, 000, 000, 000
Output
对于每一组 op = 2 的询问,输出一行一个值表示答案。
Sample Input
4 4
2 3 4 5
1 2
2 3
2 4
0 3 4
0 1 3
1 2 3
1 1 4
Sample Output
2
4
破题:看到树不要慌 建树最重要做了两天树,要会写dfs();写出来以后这个树还不是任你操作了??
不要怜惜那内存!!;
思路:
dfs建树
struct node只有一个元素但是是个模板可以多次用的!!;
反正就是把一个无向图从1这个节点提起来顺着给的数据往后处理的;
还有 函数 f();里面的要点是对节点处理不要忘记头尾和结束的点,可以输出看看有没有记录成功;
没有要求顺序之类的 就一下子压到栈里不漏就行;
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=100010;
struct node{
int nt;
};
vector<node>v[maxn];
bool book[maxn];
int dth[maxn],pre[maxn],val[maxn];
int qwq[maxn];
void dfs(int x)
{
book[x]=1;
for(int i=0;i<v[x].size();i++)
{
if(book[v[x][i].nt])continue;
pre[v[x][i].nt]=x;
dth[v[x][i].nt]=dth[x]+1;
dfs(v[x][i].nt);
}
}
int f(int q,int x,int y){
int cnt=0,flag=0;
while(x!=y)
{
if(dth[x]>=dth[y]){
qwq[cnt++]=x;
x=pre[x];
flag=0;
}else{
qwq[cnt++]=y;
y=pre[y];
flag=1;
}
}
if(flag==0){
qwq[cnt++]=x;
}else{
qwq[cnt++]=y;
}
if(q==0)
{
for(int i=0;i<cnt;i++){
val[qwq[i]]=sqrt(val[qwq[i]]);
}
}else{
ll sum=0;
for(int i=0;i<cnt;i++){
//cout<<val[qwq[i]]<<endl;
sum+=val[qwq[i]];
}
cout<<sum<<endl;
}
return 0;
}
int main()
{
int n,q;
memset(book,0,sizeof(book));
scanf("%d%d",&n,&q);
for(int i=0;i<=n;i++)v[i].clear();
int x,y;
node a;
for(int i=1;i<=n;i++)
{
cin>>val[i];
}
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&x,&y);
a.nt=y;
v[x].push_back(a);
a.nt=x;
v[y].push_back(a);
}
dfs(1);
for(int i=0;i<q;i++){
int qu,a,b;
cin>>qu>>a>>b;
f(qu,a,b);
}
return 0;
}