Codeforces ~ 1082D ~ Maximum Diameter Graph （构造，树）

题意

N个点，给定每个点的最大值，要求构造一棵树，使得直径最长。

思路

首先度数超过2的都是直径上的点，剩下的度为1的点往上面挂即可，注意先在第一个和最后一个挂上一个。
什么时候是NO呢，除了根节点和叶子结点，每个点度数至少为2，叶子最少为1个。那么度数和最小为2*n-2。

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 505;
int n, a[MAXN];
vector<int> A, B;
vector<pair<int, int> > ans;
int main()
{
    scanf("%d", &n);
    int sum = 0;
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
        if (a[i] == 1) A.push_back(i);
        else B.push_back(i);
        sum += a[i];
    }
    if (sum < 2*n-2) { printf("NO\n"); return 0; }
    int d = B.size()-1;
    for (int i = 1; i < B.size(); i++)
    {
        ans.emplace_back(B[i - 1], B[i]);
        a[B[i - 1]]--, a[B[i]]--;
    }
    if (A.size() >= 1)
    {
        ans.emplace_back(A[0], B[0]);
        a[A[0]]--, a[B[0]]--;
    }
    if (A.size() >= 2)
    {
        ans.emplace_back(B.back(), A[1]);
        a[B.back()]--, a[A[1]]--;
    }
    int cnt = 0;
    for (int i = min(2, (int)A.size()); i < A.size(); i++)
    {
        while (a[B[cnt]] == 0) cnt++;
        ans.emplace_back(B[cnt], A[i]);
        a[B[cnt]]--, a[A[i]]--;
    }
    printf("YES %d\n", B.size() - 1 + min(2, (int)A.size()));
    printf("%d\n", ans.size());
    for (auto i: ans) printf("%d %d\n", i.first + 1, i.second + 1);
    return 0;
}
/*
3
2 2 2
*/