codeforces 1003 E. Tree Constructing(构造)

题目链接:http://codeforces.com/contest/1003/problem/E

思路:先构造一条长度为d+1的链,每条链可延伸的深度是知道的,可以DFS,DFS的时候顺便可以检验一下是否可行,在注意一下k=1这些特殊情况判断。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
ll n,d,k,s,x,y;
vector<Pll>g;
void dfs(ll x,ll z,ll k,ll q,ll qw)
{
    if(z==k||!n)  return;
    while(q--)
    {
        if(!n)  return;
        g.pb(Pll(x,y));
        y++,n--;
        dfs(y-1,z,k+1,qw-1,qw);
    }
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n>>d>>k;
    if((n>2&&k<2)||(n==1&&k==1)||n<=d)  return puts("NO"),0;
    k-=2ll,d++,n-=d,y=d+1;
    FOR(i,2,d-1)
    {
        if(n<=0)  break;
        if(i<=(d+1ll)/2ll)  x=i-1;
        else  x=d-i;
        dfs(i,x,0,k,k+2);
    }
    if(n)  return puts("NO"),0;
    puts("YES");
    FOR(i,1,d-1)   printf("%d %d\n",i,i+1);
    for(int i=0;i<g.size();i++)   printf("%I64d %I64d\n",g[i].first,g[i].second);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40858062/article/details/80987963