codeforces round#408(div.2) C. Bank Hacking

来源:互联网 发布:popper.js是什么 编辑:程序博客网 时间:2024/06/03 09:45

题意:相当于若干个银行构成了一个无根树,我们需要把所有的银行hack掉,刚开始可以随意选一个,后来就只能选被hack过的邻居才行,每个银行都有他们的strength,hack它需要不比它小的strength值,每次hack一个银行后,它的邻居和邻居的邻居strength值都会+1,求hack全部的银行需要的最小strength值。


可以想,如果只有一个最大值,那么存在两种情况,如果所有银行的strength值-1的值都连在了这个最大值上,那么所需最大strength值就是这个最大值,反之所需最大strength值为这个最大值+1。

如果有多个最大值,也存在两种情况,如果存在一个点使得所有最大值到该点的距离<=1,那么所需strength值为这些银行最大值+1,反之所需strength值为最大值+2。

#include<bits/stdc++.h>using namespace std;int n,a[300005],ma=-1e9,man=0,ma_n=0,u,v,x;vector<int>G[300005];int main(){    cin>>n;    for(int i=1;i<=n;i++)    {        cin>>a[i];        ma=max(ma,a[i]);    }    for(int i=1;i<=n;i++)    {        if(a[i]==ma)man ++;        if(a[i]==ma-1)ma_n++;    }    for(int i=0;i<n-1;i++)    {        scanf("%d%d",&u,&v);        G[u].push_back(v);        G[v].push_back(u);    }    if(man==1)    {        int k,sum=0;        for(k=1;k<=n;k++)        {            if(a[k]==ma)break;        }        for(int i=0;i<G[k].size();i++)        {            if(a[G[k][i]]==ma-1)sum++;        }        if(sum==ma_n)        {            cout<<ma<<endl;            return 0;        }        cout<<ma+1;        return 0;    }    else    {        int k,sum=0;        for(k=1;k<=n;k++)        {            if(a[k]==ma)break;        }        for(int i=0;i<G[k].size();i++)        {            if(a[G[k][i]]==ma)sum++;        }        if(sum==man-1)        {            cout<<ma+1;            return 0;        }        for(int i=0;i<G[k].size();i++)        {            sum=0;x=G[k][i];            if(a[x]==ma)sum++;            for(int j=0;j<G[x].size();j++)            {                if(a[G[x][j]]==ma)sum++;            }            if(sum==man)            {                cout<<ma+1;                return 0;            }        }    }    cout<<ma+2;    return 0;}


1 0
原创粉丝点击