Codeforces Round #527 (Div. 3) E. Minimal Diameter Forest

链接

https://codeforces.com/problemset/problem/1092/E

题解

花了一晚上写这道题(时间都用来调试$py$代码了)
对每个连通分量找到一个点，使得这个点到其它点的距离的最大值最小，由于$n\le 1000$可以直接暴力$O(n^2)$
对于第$i$个连通分量，记这个点为$cente{r}_i$，这个最小的最大距离为$d_i$（也就等于直径的一半）
然后找到最大的那个$d_i$对应的$i$，把其余的$cente{r}_j$都直接连到$cente{r}_i$上
为啥这样就最优？
记原来最远的两个点的距离为$D$
当$D$为偶数时：
那么最大的$d_i$等于$\frac{D}{2}$，如果第二小的$d_i$小于$\frac{D}{2}$，那么我构造出的树的直径是$D$，是最优解
如果第二小的$d_i$也等于$\frac{D}{2}$，第三小的$d_i$小于$\frac{D}{2}$，我构造出的树直径是$D+1$
现在看一下有没有可能存在直径为$D$的答案呢？
假设这条直径的两个端点为$u,v$，假设那棵$d_i$也等于$\frac{D}{2}$的树上离根节点距离最远的点是$t$，那么不管怎么连，$u$肯定最后和$t$是连通的，显然想要构造尽量小的直径，我肯定直接把$u$所在的$component$上的某个点直接用一条边连接到$t$所在的$component$而不是经过多次中转which will lead to bigger answer.那么如果我在$u$所在的$conpoment$上的点不是距离它恰好$\frac{D}{2}$的点，肯定答案会更劣，$t$所在的分量中的点的选择可以用类似的方法证明，而从一个连通分量到另一个连通分量至少要经过$1$的距离，所以答案不会小于$D+1$
如果有三个及以上分量的$d_i$都等于$\frac{D}{2}$，那么答案是$D+2$
因为有一个分量通过走两条边到达了另一个分量，而因为这是树这样的情况是无法避免的
当$D$为奇数时
证明是类似的

温馨提示

$ptyhon$代码调试了一晚上
期间还学会了怎么$try,catch$
最后发现$python$的递归层数的最大值默认是$1000$，而这道题恰好有$1000$个点
所以就用了这个东西
sys.setrecursionlimit(n)
其中$n$表示你要把递归层数的上限设为多少

代码

import sys
sys.setrecursionlimit(1100)

def dfs1(u,pre):            #find the components
    vis[u] = True
    now.append(u)
    for v in to[u]:
        if v!=pre:dfs1(v,u)

def dfs2(u,pre):            #calulate the distance
    mxdist[u]=dist[u]
    for v in to[u]:
        if v!=pre:
            dist[v] = dist[u]+1
            dfs2(v,u)
            mxdist[u]=max(mxdist[u],mxdist[v])

try:
    lab = 1
    n, m = [int(x) for x in input().split()]
    to = [[] for i in range(n+10)]
    dist = [0 for i in range(n+10)]
    mxdist = [0 for  i in range(n+10)]

    lab = 2
    for i in range(m):
        u,v = [int(x) for x in input().split()]
        to[u].append(v)
        to[v].append(u)

    com=[]
    vis=[False for i in range(n+10)]
    for i in range(1,n+1):
        if vis[i]==False:
            now=[]
            dfs1(i,0)
            com.append(now)

    lab = 3
    ct = []
    mx = 0
    for lis in com:
        tmp = []
        d = 0
        for root in lis:
            for u in lis:dist[u]=mxdist[u]=0
            dfs2(root,0)
            tmp.append((mxdist[root],root))
            d = max( d, sum( sorted([ mxdist[u] for u in to[root] ])[-2:] ) )
            #print(*[mxdist[u] for u in lis])
        mx = max(mx,d)
        #print('d =',d)
        for x in tmp:
            if x[0]==(d+1)//2:
                center = [x[1] for x in tmp if x[0]==(d+1)//2][0]
        ct.append( ((d+1)//2,center) )

    #print(*ct)

    lab = 4
    ct.sort(reverse=True)
    ans = []
    for i in range(1,len(ct)):
        mx = max(mx,ct[i][0]+1+ct[0][0])
        if i>1:mx = max(mx,ct[i][0]+2+ct[1][0])
        ans.append((ct[i][1],ct[0][1]))
    print(mx)
    for p in ans:
        print(*p)
except Exception as e:
    print('error after lable',lab,', type =',e)