POJ3417 Network – 树上差分 – lca
题意:给出一个无向图,分别给出n-1条树边(主要边)和m条非树边(附加边),这个无向图可以看做一棵树外加m条附加边,你可以切断一条主要边和一条附加边,求切割后,能够使这个无向图不再连通的切割方案数(即使只切断一条主要边就可以使图不连通,你也需要再切断一条附加边)
我们先考虑只有一条附加边(x,y)时,这时这张图就是一棵基环树 我们发现如果x,y之间有一条附加边,则这条边和x到y的路径组成了一个环,如果说我们要切割x到y的路径上的一条主要边,我们必须要再切断这条附加边,才能使图不再连通
那么如果x,y之间有两条或以上附加边,若我们已切割了x到y路径上的一条主要边,那么是无法通过仅再切割一条附加边来使图不再连通
因而我们每次读入一条附加边,就给x到y的路径上的所有主要边记录上“被覆盖一次”,这样再去遍历所有主要边 对于我们想要切割的一条主要边,有以下3种情况
若这条边被覆盖0次,则可以任意再切断一条附加边 若这条边被覆盖1次,那么只能再切断唯一的一条附加边 若这条边被覆盖2次及以上,没有可行的方案 现在的问题是如何快速求出每条边被覆盖了多少次,对于这类问题,可以类比序列差分,有树上差分算法
设差分数组dif初值为0,若x,y有一条附加边,则dif[x]++,dif[y]++,dif[lca(x,y)]-=2 设f(x)为以x为根的子树中所有节点dif之和,则f(x)就是x到其父节点的边被覆盖的次数 没错,求的是子树和,所以说求各种类似前缀和的东西真的很好用,一般可以用这些方式对区间O(1)求解
为了更快地解决问题,有必要总结各种思想与算法,并分析其适用于什么样的情况,要时常复习自己的博客啊… 比如说这题给我的启示是,树上问题可以类比序列问题,比如说求树上s到t点的路径和,就可以用树上前缀和 最后处理答案的时候注意,若f[x] == 0,则x可能为根节点,要特判,根节点没有父节点!!! 一般我们记录一个点“上面的一条边”
特别注意,LCA若是循环从20开始,则数组要开到21以上!不然会越界RE #include #include #include #include #include using namespace std; #define debug(x) cerr << #x << "=" << x << endl; const int MAXN = 100000 + 10; const int INF = 1<<30; int n,m,tot,last[MAXN],f[MAXN][21],d[MAXN],dif[MAXN],vis[MAXN],ans,sta[MAXN]; struct Edge { int u,v,w,to; Edge(){} Edge(int u, int v, int to) : u(u), v(v), to(to) {} }e[MAXN * 2]; inline void add(int u, int v) { e[++tot] = Edge(u,v,last[u]); last[u] = tot; } void lca_dfs(int now) { for(int i=last[now]; i; i=e[i].