[校内模拟][树形DP] 201016 Practice T2 骑行Ride

T2

一棵树，求不同的有向单峰路径数量
做法很多我都不会

手玩之后发现

$$ans=(\sum s_i{)}^2-\sum s_i^2$$
其中$s_i$指子树中单调下降路径的数量
可以用乘法分配律解释

然后两行代码就完了😢

#include<bits/stdc++.h>
using namespace std;
#define in Read()
int in{
	int i=0,f=1;char ch=0;
	while(!isdigit(ch)&&ch!='-') ch=getchar();
	if(ch=='-') ch=getchar(),f=-1;
	while(isdigit(ch)) i=(i<<3)+(i<<1)+ch-48,ch=getchar();
	return i*f;
}

const int N=3e5+5;
int n;
long long ans;
int tot,first[N],nxt[N<<1],aim[N<<1];
int f[N];

void ljb(int u,int v){
	++tot;
	nxt[tot]=first[u];
	first[u]=tot;
	aim[tot]=v;
	return;
}

int main(){
	n=in;
	for(int i=1;i<n;++i){
		int u=in,v=in;
		ljb(u,v);
		ljb(v,u);
	}
	for(int u=1;u<=n;++u){
		for(int e=first[u];e;e=nxt[e]){
			int v=aim[e];
			if(v>u) continue;
			f[u]+=f[v]+1;
		}
		for(int e=first[u];e;e=nxt[e]){
			int v=aim[e];
			if(v>u) continue;
			ans+=1ll*(f[u]-f[v]-1)*(f[v]+1);
		}
	}
	cout<<ans;
	return 0;
}