codeforces 796c Bank Hacking

题意：

有很多家银行，如果一个上线的银行和另一个上线的银行直接相连，那么称这种关系叫相邻；

如果一个上线的银行和另一个上线的银行通过第三个上线银行间接相连，称这种情况为半相邻。

每个银行一开始是上线的并且有一个初始的防卫力量（可能为负数），一个银行被抢劫之后就会下线并且再也不会上线，与它相邻和半相邻的所有上线的银行防卫力量都加一。

一个人要去抢劫银行，首先他会选择一家银行进行抢劫，之后抢劫的银行需要满足以下条件：

1.这家上线的银行与一家下线的银行直接相邻；

2.这家银行的防守值不能超过电脑的攻击值。

问电脑的攻击值最小为多少可以保证抢劫完所有的银行。

思路：

首先理清题意，被抢劫的银行必须与一家下线的银行直接相邻。这个条件十分重要，它保证了每家银行的防守值要么加1，要么加2。

所以，如果一家银行的防守值为x，与它相连的所有银行中的防守值最大为y，不与它相连的所有银行中防守值最大的为z，那么第一次抢劫这家银行的答案就是x，y+1，z+2中的最大值。

所以第一种方法就出来啦，枚举所有的银行作为第一次的抢窃，然后算x，y，z。

这种方法稍不注意就会tle，但是巧妙的利用multiset就可以完美解决这个问题，复杂度为nlog(n)，但是常数较大。

第二种方法就对思维的要求比较高。

设防守值的最大为mx，

那么统计与每一个点直接相连的mx和mx-1的点的数量。

有一下三个情况：

1.只有一个mx，并且这个mx与所有的mx-1直接相邻，这时答案就是mx；

2.一个mx与所有的mx直接相邻，这时的答案是mx+1；

3.其他情况mx+2。

需要好好想一想。

方法1代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <set>
using namespace std;

multiset<int> mmp;
const int inf = 0x3f3f3f3f;
const int N = 3e5 + 10;
int a[N];
vector<int> g[N];

int main()
{
    int n;
    
    scanf("%d",&n);
    
    for (int i = 1;i <= n;i++)
    {
        scanf("%d",&a[i]);
        mmp.insert(a[i]);
    }
    
    for (int i = 0;i < n - 1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    
    int ans = inf;
    
    for (int i = 1;i <= n;i++)
    {
        int tmp = a[i];
        mmp.erase(mmp.find(a[i]));
        for (int j = 0;j < g[i].size();j++)
        {
            int k = a[g[i][j]];
            mmp.erase(mmp.find(k));
            tmp = max(k+1,tmp);
        }
        if (!mmp.empty()) tmp = max(tmp,*(--mmp.end()) + 2);
        ans = min(tmp,ans);
        mmp.insert(a[i]);
        for (int j = 0;j < g[i].size();j++)
        {
            int k = a[g[i][j]];
            mmp.insert(k);
        }
    }
    
    printf("%d\n",ans);
    
    return 0;
}

方法2代码：

#include <stdio.h>
#include <string.h>
#include <vector>
#include <algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 3e5 + 10;
int a[N];
vector<int> g[N];

int main()
{
    int n;
    
    scanf("%d",&n);
    
    int maxn = -inf;
    
    for (int i = 1;i <= n;i++)
    {
        scanf("%d",&a[i]);
        maxn = max(maxn,a[i]);
    }
    
    for (int i = 0;i < n - 1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    
    int mxd = 0,mx = 0;
    
    for (int i = 1;i <= n;i++)
    {
        if (a[i] == maxn) mx++;
        if (a[i] == maxn - 1) mxd++;
    }
    
    bool c = 0,d = 0,e = 0;
    
    if (mx == 1)
    {
        for (int i = 1;i <= n;i++)
        {
            if (a[i] == maxn)
            {
                int cnt = 0;
                for (int j = 0;j < g[i].size();j++)
                {
                    int x = a[g[i][j]];
                    
                    if (x == maxn - 1) cnt++;
                }
                if (cnt == mxd) c = 1;
            }
        }
        
        if (c) return 0*printf("%d",maxn);
    }
    
    for (int i = 1;i <= n;i++)
    {
        int cnt = 0;
        if (a[i] == maxn) cnt++;
        for (int j = 0;j < g[i].size();j++)
        {
            if (a[g[i][j]] == maxn) cnt++;
        }
        if (cnt == mx) return 0*printf("%d",maxn + 1);
    }
    
    printf("%d",maxn + 2);
    
    return 0;
}