题意
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
*/