傳送門

題意：

路徑有$-1,1$兩種權值，求有多少路徑滿足權值和為$0$且有一個點將路徑分成權值和為$0$的兩段

---

?

第四節課本來想去上化學，然后快上課了這道題還沒調出來.....可惡我想上化學

昨天兩節語文課潸然的李煜講座也沒去聽嗚嗚聽說今天的語文課還有什么文藝活動又錯過了嗚嗚

還是有思路的

點分治，考慮經過$u$的路徑

首先保證權值和為$0$，記錄$c[i]$為當前權值和為$i$的路徑有幾條

怎么滿足有一個點呢？

$0+0=0$!!!

我們只要保證一段和$0$另一段自然也是$0$

所以補充保存的信息，

$c[i][0]$當前權值和為$0$且到$u$的路徑上沒有一段為$0$(就是沒有祖先的權值和也為$i$，打標記就行了)的路徑有幾條

$c[i][1]$表示有......

然后每顆子樹先更新答案再更新$c$就行了

然后一直$WA$.....

突然想到自己沒有處理$u$的信息

然后改了也不對

然后發現打標記只是$=1\ =0$沒有$++\ --$.....

改了還不對

參考黃學長的代碼打了一份他的做法$AC$了，然后又開始找自己的錯誤

最后還是處理$u$的信息有問題，我直接用了$mark$因為我還以為$mark$只有$0,1$.......

還我化學

?

#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; typedef long long ll; const int N=1e5+5,INF=1e9+5; inline int read(){char c=getchar();int x=0,f=1;while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}return x*f; } int n,a,b,w; struct edge{int v,w,ne; }e[N<<1]; int h[N],cnt; inline void ins(int u,int v,int w){cnt++;e[cnt].v=v;e[cnt].w=w;e[cnt].ne=h[u];h[u]=cnt;cnt++;e[cnt].v=u;e[cnt].w=w;e[cnt].ne=h[v];h[v]=cnt; }int f[N],size[N],all,vis[N],root; void dfsRt(int u,int fa){size[u]=1;f[u]=0;for(int i=h[u];i;i=e[i].ne){int v=e[i].v;if(vis[v]||v==fa) continue;dfsRt(v,u);size[u]+=size[v];f[u]=max(f[u],size[v]);}f[u]=max(f[u],all-size[u]);if(f[u]<f[root]) root=u; } int mark[N<<1],Z=N,c[N<<1][2]; ll ans; void dfsAns(int u,int fa,int now){if(now==0) ans+=c[Z][0]+c[Z][1] +(mark[Z]>=1);else{if(mark[Z+now]) ans+=c[Z-now][0]+c[Z-now][1];else ans+=c[Z-now][1];}mark[Z+now]++;for(int i=h[u];i;i=e[i].ne)if(!vis[e[i].v]&&e[i].v!=fa) dfsAns(e[i].v,u,now+e[i].w);mark[Z+now]--; } int st[N],top; void dfsDee(int u,int fa,int now){c[Z+now][mark[Z+now]>=1]++;st[++top]=Z+now;mark[Z+now]++;for(int i=h[u];i;i=e[i].ne)if(!vis[e[i].v]&&e[i].v!=fa) dfsDee(e[i].v,u,now+e[i].w);mark[Z+now]--; } void dfsSol(int u){vis[u]=1;for(int i=h[u];i;i=e[i].ne)if(!vis[e[i].v]) dfsAns(e[i].v,u,e[i].w),dfsDee(e[i].v,u,e[i].w);while(top) c[st[top]][0]=c[st[top]][1]=0,top--;for(int i=h[u];i;i=e[i].ne) if(!vis[e[i].v]){all=size[e[i].v];root=0;dfsRt(e[i].v,0);dfsSol(root);} } int main(){freopen("in","r",stdin);n=read();for(int i=1;i<n;i++) a=read(),b=read(),w= read()==0?-1:1,ins(a,b,w);all=n;root=0;f[0]=INF;dfsRt(1,0);dfsSol(root);printf("%lld",ans); }

?

黃學長做法，本質上一樣的

改進了一下后$1600ms$

#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; typedef long long ll; const int N=1e5+5,INF=1e9+5; inline int read(){char c=getchar();int x=0,f=1;while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}return x*f; } int n,a,b,w; struct edge{int v,w,ne; }e[N<<1]; int h[N],cnt; inline void ins(int u,int v,int w){cnt++;e[cnt].v=v;e[cnt].w=w;e[cnt].ne=h[u];h[u]=cnt;cnt++;e[cnt].v=u;e[cnt].w=w;e[cnt].ne=h[v];h[v]=cnt; }int f[N],size[N],all,vis[N],root; void dfsRt(int u,int fa){size[u]=1;f[u]=0;for(int i=h[u];i;i=e[i].ne){int v=e[i].v;if(vis[v]||v==fa) continue;dfsRt(v,u);size[u]+=size[v];f[u]=max(f[u],size[v]);}f[u]=max(f[u],all-size[u]);if(f[u]<f[root]) root=u; } int mark[N<<1],Z=N; ll c[N<<1][2],d[N<<1][2]; ll ans; int mx; void dfsRun(int u,int fa,int now){d[Z+now][mark[Z+now]>=1]++;mx=max(mx,abs(now));mark[Z+now]++;for(int i=h[u];i;i=e[i].ne)if(!vis[e[i].v]&&e[i].v!=fa) dfsRun(e[i].v,u,now+e[i].w);mark[Z+now]--; } void dfsSol(int u){//printf("dfsSol %d\n",u);vis[u]=1;c[Z][0]=1;int mxmx=0;for(int i=h[u];i;i=e[i].ne)if(!vis[e[i].v]){mx=1;dfsRun(e[i].v,u,e[i].w);mxmx=max(mxmx,mx);ans+=(c[Z][0]-1)*d[Z][0];for(int j=-mx;j<=mx;j++) ans+=c[Z-j][1]*d[Z+j][1]+c[Z-j][0]*d[Z+j][1]+c[Z-j][1]*d[Z+j][0];for(int j=Z-mx;j<=Z+mx;j++){c[j][0]+=d[j][0];c[j][1]+=d[j][1];d[j][0]=d[j][1]=0;}}for(int i=Z-mxmx;i<=Z+mxmx;i++) c[i][0]=c[i][1]=0;for(int i=h[u];i;i=e[i].ne) if(!vis[e[i].v]){all=size[e[i].v];root=0;dfsRt(e[i].v,0);dfsSol(root);} } int main(){freopen("in","r",stdin);n=read();for(int i=1;i<n;i++) a=read(),b=read(),w= read()==0?-1:1,ins(a,b,w);all=n;root=0;f[0]=INF;dfsRt(1,0);dfsSol(root);printf("%lld",ans); }

?

轉載于:https://www.cnblogs.com/candy99/p/6497858.html

總結

以上是生活随笔為你收集整理的BZOJ 3697: 采药人的路径 [点分治] [我想上化学课]的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。