T2
一棵树,求不同的有向单峰路径数量
做法很多我都不会
手玩之后发现
a
n
s
=
(
∑
s
i
)
2
−
∑
s
i
2
ans=(\sum s_i)^2-\sum s_i^2
ans=(∑si)2−∑si2
其中
s
i
s_i
si指子树中单调下降路径的数量
可以用乘法分配律解释
然后两行代码就完了😢
#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;
}