BZOJ 3712: [PA2014]Fiolki 倍增+想法

3712: [PA2014]Fiolki

Time Limit: 30 Sec  Memory Limit: 128 MB
Submit: 437  Solved: 115
[Submit][Status][Discuss]

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

HINT

 

Source

鸣谢Jcvb

想法：

先看出树结构，然后分析：

一个反应最多进行一次，而且反应一定是在其LCA处反应。先不考虑中途其他反应，那么这个反应进行的时间是可以知道的。将这些反应按（反应时间，优先顺序）sort一下。模拟即可。

求反应时间就是求两个点最慢到LCA的时间，可以倍增。O(nlogn+klogn)

还有一种方法，按操作顺序，每操作一次建一个新容器用来装反应完的产物，例：x->y 这样： x->tmp y->tmp 然后y=tmp用来和后面的反应。

这样节点的深度与反应时间反相关，就可以不用倍增求，而用Tarjan求。从而O(n+k)

附上倍增代码

#include<cstdio>
#include<cmath>
#include<algorithm>

typedef long long ll;
const int len(),lem(),INF(0x7fffffff);
int n,m,k,g[len+],a,b,logg;
int f[][len+],t[len+],depth[len+];
struct AB{int a,b,lca,id;}reaction[lem+];
struct Node{int nd,nx;}bot[len+];int tot,first[len+];
ll precipitation,ans;

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
bool cmp(AB A,AB B){return (A.lca<B.lca)||(A.lca==B.lca&&A.id<B.id);}
void add(int a,int b){bot[++tot]=(Node){b,first[a]};first[a]=tot;}
void swap(int &a,int &b){if(a==b)return;a^=b;b^=a;a^=b;}
template <class T>void read(T &x)
{
    x=;int f=;char c=getchar();
    while((c<''||c>'')&&c!='-')c=getchar(); if(c=='-')f=,c=getchar();
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    x=f?-x:x;
}

void DFS(int x)
{
    depth[x]++;
    for(int v=first[x];v;v=bot[v].nx)
    depth[bot[v].nd]=depth[x],DFS(bot[v].nd);
}
int lca(int a,int b)
{
    if(depth[a]<depth[b])swap(a,b);
    for(int k=depth[a]-depth[b]-,j=;k>;k>>=,j++)
    if(k&)a=f[j][a];
    if(f[][a]==b)return t[a];
    if(depth[a]>depth[b]) a=f[][a];
    for(int j=logg;f[][a]!=f[][b];j--)
    if(f[j][a]!=f[j][b])a=f[j][a],b=f[j][b];
    return max(t[a],t[b]);
}
int main()
{
//    freopen("C.in","r",stdin);
//    freopen("C.out","w",stdout);
    read(n),read(m),read(k);
    for(int i=;i<=n;i++)read(g[i]);
    for(int i=;i<=m;i++)
    {
        read(a),read(b);
        f[][a]=b; t[a]=i;    add(b,a);
    }
    for(int i=;i<=n;i++)if(!f[][i])f[][i]=n+,add(n+,i),t[i]=INF;
    DFS(n+);
    logg=log2(n);
    for(int j=;j<=logg;j++)
       for(int i=;i<=n;i++)
       f[j][i]=f[j-][ f[j-][i] ];
    for(int i=;i<=k;i++)
    {
        read(a),read(b);
        reaction[i].a=a;
        reaction[i].b=b;
        reaction[i].lca=lca(a,b);
//        if(!reaction[i].lca)fprintf(stderr,"%d\n",i);
        reaction[i].id=i;
    }
    std::sort(reaction+,reaction++k,cmp);
    for(int i=;i<=k;i++)
    if(reaction[i].lca!=INF)
    {
        precipitation=min(g[reaction[i].a],g[reaction[i].b]);
        ans+=precipitation;
        g[reaction[i].a]-=precipitation;
        g[reaction[i].b]-=precipitation;
    }else break;
    printf("%lld",ans<<);
    return ;
}