将每个点到根节点的最短路给出来,然后限定每个节点和其他节点相关联的边不超过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()
{
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;
int knum=du[i].size(),q=0;
for(int p=0;p<knum;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;
}