hdu2376-Average distance（树上平均距离）+ cf429A-Xor-tree

hdu2376-Average distance

题意

求数上任意两点之间距离的总和并计算其平均值。

两种方法：①求每条边的贡献，即这条边的两侧分别有多少端点，则该边在答案中被计算的次数即为两侧端点的数量的乘积。

②直接暴力每个点到其他点的距离，由于会重复计算，除 $2$ 后即与方法①中得到的结果相同。

#include <bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;

const int N=1e4+10;
vector<PII> h[N];
int cnt[N];
int sum=0;
int n;

void dfs(int u,int fa){
	cnt[u]=1;
	for(auto x:h[u]){
		if(x.first==fa)
			continue;
		dfs(x.first,u);
		cnt[u]+=cnt[x.first];
	}
}

void dfs1(int u,int fa){
	if(h[u].size()==1&&u!=0)//叶子结点 没有再往下的边了 
		return ;
	for(auto x:h[u]){
		if(x.first==fa)
			continue;
		dfs1(x.first,u);
		sum+=(x.second*cnt[x.first]*(n-cnt[x.first]));//该边乘上下两侧的点个数 
	}
}

void solve(){
	sum=0;
	memset(cnt,0,sizeof(cnt));
	cin>>n;
	for(int i=0;i<n;i++)
		h[i].clear();
	int a,b,c;
	for(int i=1;i<n;i++){
		cin>>a>>b>>c;
		h[a].push_back({b,c});
		h[b].push_back({a,c});
	}
	dfs(0,-1);
	dfs1(0,-1);
	double ans=sum*1.0/(n*(n-1)*1.0/2);
	printf("%.8lf\n",ans);
}

signed main(){
	int t;
	cin>>t;
	while(t--)
		solve();
	return 0;
}

cf429A-Xor-tree

题意

给定一棵以 $1$ 为根节点，且所有结点的值只能为 $0$ 或 $1$ 的树。每次操作可以选中一个结点对其值进行翻转，同时可对其孙子，孙子的孙子等进行翻转（隔代操作）。初始时每个编号的状态为 $a_i$ ，现需要让每个编号的状态在若干次操作后变为 $b_i$ ，求最少的操作次数及操作的结点。

显然对于需要翻转的某点，若其祖先中所有隔代点都不需要操作，那么直接贪心的翻转该点即可（因为都会对孙子中的隔代产生相同的效果），同时更新孙子的状态。

由于翻转偶数次=没翻，翻转奇数次=翻转，因此对于记录翻转次数的 $st$ ，直接在 $0$ 和 $1$ 中进行转换即可。因为只对隔代产生贡献，考虑在每次 $dfs$ 中多开一个 $st$ 状态交换进行。

#include <bits/stdc++.h>
#define int long long
#define PII pair<int,int>
using namespace std;

const int N=1e5+10;

vector<int> h[N];
int flag[N];
int a[N],b[N];
int sum=0;
vector<int> ans;

void dfs(int u,int fa,int st1,int st2) {
	if((a[u]^st2)!=b[u]){
		st2^=1;
		ans.push_back(u);
	}
	for(auto x:h[u]) {
		if(x==fa)
			continue;
		dfs(x,u,st2,st1);//此时dfs中交换了st1和st2
	}
}

signed main(){
	int n,x;
	cin>>n;
	int u,v;
	for(int i=1;i<n;i++){
		cin>>u>>v;
		h[u].push_back(v);
		h[v].push_back(u);
	}
	
	for(int i=1;i<=n;i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
		cin>>b[i];
	
	dfs(1,0,0,0);
	cout<<ans.size()<<endl;
	for(auto x:ans)
		cout<<x<<endl;
	
	return 0;
}