[BZOJ3712]Fiolki 重构树（并查集）

3712: [PA2014]Fiolki

Time Limit: 30 Sec  Memory Limit: 128 MB

Description

化学家吉丽想要配置一种神奇的药水来拯救世界。
吉丽有n种不同的液体物质，和n个药瓶（均从1到n编号）。初始时，第i个瓶内装着g[i]克的第i种物质。吉丽需要执行一定的步骤来配置药水，第i个步骤是将第a[i]个瓶子内的所有液体倒入第b[i]个瓶子，此后第a[i]个瓶子不会再被用到。瓶子的容量可以视作是无限的。
吉丽知道某几对液体物质在一起时会发生反应产生沉淀，具体反应是1克c[i]物质和1克d[i]物质生成2克沉淀，一直进行直到某一反应物耗尽。生成的沉淀不会和任何物质反应。当有多于一对可以发生反应的物质在一起时，吉丽知道它们的反应顺序。每次倾倒完后，吉丽会等到反应结束后再执行下一步骤。
吉丽想知道配置过程中总共产生多少沉淀。

Input

第一行三个整数n,m,k(0<=m<n<=200000,0<=k<=500000)，分别表示药瓶的个数（即物质的种数），操作步数，可以发生的反应数量。
第二行有n个整数g[1],g[2],…,g[n]（1<=g[i]<=10^9)，表示初始时每个瓶内物质的质量。
接下来m行，每行两个整数a[i],b[i](1<=a[i],b[i]<=n,a[i]≠b[i])，表示第i个步骤。保证a[i]在以后的步骤中不再出现。
接下来k行，每行是一对可以发生反应的物质c[i],d[i](1<=c[i],d[i]<=n,c[i]≠d[i])，按照反应的优先顺序给出。同一个反应不会重复出现。

Output

Sample Input

3 2 1
2 3 4
1 2
3 2
2 3

Sample Output

6

 

 

 

考试的时候，上来凭直觉觉得是并查集……

此后第a[i]个瓶子不会再被用到.

怎么看怎么是并查集嘛，然后……打了个链表的O(玄学)暴力,就开始想怎么优化查找的过程

可是到考试结束也没想出来-_-最后只好交暴力了

回了宿舍Troywardalao问我题面是什么(考试时候他都没看题,然而考完试他10min就淼掉了233)

然后两个人想了10min就做出来了...(Torywar dalao为什么你想正解想得这么熟练啊)

我们考虑这个倒药品的过程,很显然倒药品的顺序会产生影响.

如果我们按照普通的并查集一样合并,会产生一系列问题,最大的问题就是不知道有哪些反应

我们考虑每次合并新建节点并向他们连边(也就是重构树,参见http://www.cnblogs.com/LadyLex/p/7275821.html)

这样,我们就把每一次合并拆开,并且独立考虑每次合并的过程

接下来使用了类似离线的思想,我们首先处理一个计算lca的方式(tarjan,倍增随意了),然后考虑:

对于每一对可以反应的物质,他们可能,也只可能在他们重构树上lca处反应

既然这样,我们就可以把每个反应添加到他们的lca上

随后,在扫一遍重构时新添加到点,也就是执行一遍倒药品的过程,并且处理可能发生的反应

总的复杂度是O(m+nlogn+klogn+k),分别对应重构,ST表预处理,向lca添加反应,处理合并过程

(ps:其实还是有一点遗憾,明明自己已经打过重构树的题了,考试时候却没想到.看来我对知识的掌握程度还没有到那么熟练的程度....联赛之前一定要把已经学的知识掌握扎实啊...)

代码见下:

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <vector>
 using namespace std;
 const int N=,K=;
 typedef long long LL;
 LL sum;
 int n,m,k,w[N],fa[N<<],bin[];
 int e,adj[N<<],cnt,f[N<<][],deep[N<<];
 int find(int a){return fa[a]==a?a:fa[a]=find(fa[a]);}
 inline int min(int a,int b){return a<b?a:b;}
 struct data{int a,b;data(int x=,int y=){a=x,b=y;}}step[N];
 struct edge{int zhong,next;}s[N<<];
 vector<data>re[N<<];
 inline void add(int qi,int zhong)
     {s[++e].zhong=zhong;s[e].next=adj[qi];adj[qi]=e;}
 void dfs(int rt)
 {
     deep[rt]=deep[f[rt][]]+;
     for(int i=adj[rt];i;i=s[i].next)
         dfs(s[i].zhong);
 }
 inline void ST()
 {
     for(int i=;i<=;i++)
         for(int j=;j<=cnt;j++)
             f[j][i]=f[f[j][i-]][i-];
 }
 inline int LCA(int a,int b)
 {
     if(deep[a]<deep[b])swap(a,b);
     int cha=deep[a]-deep[b];
     for(int j=;~j;j--)
         if(cha&bin[j])a=f[a][j];
     if(a==b)return a;
     for(int j=;~j;j--)
         if(f[a][j]!=f[b][j])a=f[a][j],b=f[b][j];
     return f[a][];
 }
 int main()
 {
     bin[]=;for(int i=;i<=;i++)bin[i]=bin[i-]<<;
     int a,b,lca;scanf("%d%d%d",&n,&m,&k);cnt=n;
     for(int i=;i<=n+m;i++)fa[i]=i;
     for(int i=;i<=n;i++)scanf("%d",&w[i]);
     for(int i=;i<=m;i++)
         scanf("%d%d",&step[i].a,&step[i].b),
         a=find(step[i].a),b=find(step[i].b),
         fa[a]=fa[b]=f[a][]=f[b][]=++cnt,
         add(cnt,a),add(cnt,b);
     ST();
     for(int i=;i<=cnt;i++)
         if(!deep[i])dfs(find(i));
     for(int i=;i<=k;i++)
     {
         scanf("%d%d",&a,&b);
         if(find(a)==find(b))
             lca=LCA(a,b),re[lca].push_back(data(a,b));
     }
     for(int i=n+;i<=cnt;i++)
         for(int j=,len=re[i].size();j<len;j++)
             a=min(w[re[i][j].a],w[re[i][j].b]),
             w[re[i][j].a]-=a,w[re[i][j].b]-=a,sum+=a;
     printf("%lld\n",sum<<);
 }