CodeForces 1082 D Maximum Diameter Graph

题目传送门

题意：现在有n个点，每个点的度数最大为di，现在要求你构成一棵树，求直径最长。

题解：把所有度数为2的点先扣出来，这些就是这颗树的主干，也就是最长的距离。

然后我们把度数为2的点连起来，之后就处理1的点，先在主干的最左边和最右边加上新的点，这样可以使得直径边长。

然后其他的点随便放就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lch(x) tr[x].son[0]
#define rch(x) tr[x].son[1]
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
typedef pair<int,int> pll;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const LL mod =  (int)1e9+;
const int N = 1e5 + ;
pll p[N];
int d[N];
vector<int> vc[];
vector<pll> ans;
int main(){
    int n;
    scanf("%d", &n);
    for(int i = ; i <= n; ++i){
        scanf("%d", &d[i]);
        if(d[i] > ) vc[].pb(i);
        else vc[].pb(i);
    }
    if(vc[].size() == ){
        puts("NO");
        return ;
    }
    int fans = ;
    for(int i = ; i < vc[].size(); ++i){
        ans.pb({vc[][i], vc[][i-]});
        --d[vc[][i]], --d[vc[][i-]];
        ++fans;
    }
    int p = ;
    for(int j = ; j < vc[].size(); ++j){
        if(j == ){
            if(d[vc[][]]){
                --d[vc[][]];
                ++fans;
                ans.pb({vc[][j], vc[][]});
            }
        }
        else if(j == ){
            if(d[vc[][vc[].size()-]]){
                --d[vc[][vc[].size()-]];
                ++fans;
                ans.pb({vc[][j], vc[][vc[].size()-]});
            }
        }
        else {
            while(p < vc[].size() && d[vc[][p]] == ) ++p;
            if(p == vc[].size()) break;
            --d[vc[][p]];
            ans.pb({vc[][j], vc[][p]});
        }
    }
    if(p ==  vc[].size()) puts("NO");
    else {
        printf("YES %d\n", fans);
        printf("%d\n", ans.size());
        for(int i = ; i < ans.size(); ++i){
            printf("%d %d\n", ans[i].fi, ans[i].se);
        }
    }
    return ;
}