CodeForces 404C

将每个点到根节点的最短路给出来，然后限定每个节点和其他节点相关联的边不超过k,然后直接见图就可以了。根节点就是距离为0的点，不能有两个根节点，直接建图的时候如果发现有一个根节点必须大于k了，那就直接break了。下面是代码。

#include <iostream>
#include <cstdio>
#include <vector>
#include <stdlib.h>
#include <memory.h>
#define maxn 100010
using namespace std;

vector<int>du[maxn];
vector<int>edge[maxn];
int m,n,value,ans,k;
bool flag;
int main()
{
//    freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&m,&n))
    {
        ans=0;k=0;flag=0;
        for(int i=1;i<=m;i++){
            scanf("%d",&value);
            du[value].push_back(i);
        }
        if(du[0].size()!=1){
            printf("-1\n");continue;
        }
        for(int i=1;i<=m;i++)
        {
            int num=du[i-1].size(),j=0;//j 代表 是否是 k 叉的
            int knum=du[i].size(),q=0;//q 代表是否是 继续存在可行顶点
            for(int p=0;p<knum;p++)//p 代表有多少个点
            {
                if(i!=1&&q==num&&p<knum){flag=1;break;}
                if(j<n&&q<num){
                    j++;
                    int u=du[i-1][q],v=du[i][p];
                    edge[u].push_back(v);
                    ans++;
                }
                if(j==n){j=0;q++;}
            }
            if(flag)break;
            if(i==1)n--;
        }
        if(ans!=m-1)flag=1;
        if(flag){printf("-1\n");continue;}
        printf("%d\n",ans);
        for(int i = 1 ;i <= m ;i++)
        {
            int len = edge[i].size();
            for(int j=0 ;j<len ;j++)
            {
                printf("%d %d\n",i,edge[i][j]);
            }
        }
    }
    return 0;
}