K distance from root

PHOTO EMBED

Fri Dec 01 2023 12:17:31 GMT+0000 (Coordinated Universal Time)

Saved by @nistha_jnn #c++

void def(struct Node* root,vector<int>&ans,int k);
vector<int> Kdistance(struct Node *root, int k)
{
  vector<int>ans;
  def(root,ans,k);
  return ans;
}
void def(struct Node* root,vector<int>&ans,int k)
{
    if(!root)
    return;
    if(k==0)
    ans.push_back(root->data);
    def(root->left,ans,k-1);
    def(root->right,ans,k-1);
}
content_copyCOPY

https://www.geeksforgeeks.org/problems/k-distance-from-root/1?page=3&company=Samsung&difficulty=School,Basic,Easy&sortBy=latest