Count shortest distance in graph

Last updated: February 2, 2026

Quick Overview

Given a weighted, undirected graph represented as an adjacency list, write a function to calculate the shortest distance between two specified nodes. The function should take the graph, the starting node, and the target node as inputs, and return the shortest distance as an integer. If there is no path between the nodes, return -1.

Apple
Coding & Algorithms
Software Engineer
Apple
February 2, 2026
Software Engineer
Take-home Project
Coding & Algorithms
Medium

87

13

2,160 solved


Given a weighted, undirected graph represented as an adjacency list, write a function to calculate the shortest distance between two specified nodes. The function should take the graph, the starting node, and the target node as inputs, and return the shortest distance as an integer. If there is no path between the nodes, return -1.

Apple uses this problem in the Take-home Project to evaluate your algorithmic thinking. They expect you to discuss multiple approaches, analyze trade-offs between them, and implement the optimal solution with clean, readable code.

What the Interviewer Expects
  • Recognize the underlying problem pattern (sliding window, two pointers, BFS/DFS, etc.)
  • Discuss multiple approaches and trade-offs before coding
  • Implement an optimal solution with clean, production-quality code
  • Handle all edge cases including boundary conditions and invalid input
  • Optimize both time and space complexity with clear justification
  • Test your solution systematically with well-chosen examples
Key Topics to Cover
Dynamic programming and memoization
Time and space complexity analysis
Graph algorithms and traversal
Hash maps and frequency counting
Common algorithm patterns (sliding window, two pointers, BFS/DFS)
How to Approach This
  1. Clarify input constraints and edge cases before writing code.
  2. Walk through your approach verbally and confirm with the interviewer before coding.
  3. Start with a brute force solution, then optimize. Mention time and space complexity.
  4. Test your solution with examples, including edge cases like empty input or duplicates.
  5. Consider common patterns: sliding window, two pointers, hash map, BFS/DFS, dynamic programming.
Possible Follow-up Questions
  • How would you modify your solution to handle streaming input?
  • How would you parallelize this solution?
  • Can you optimize the space complexity of your solution?
  • Can you solve this in a single pass?
Sharpen Your Skills on Codemia

Practice similar problems with our interactive workspace, get AI feedback, and track your progress.

Practice DSA Problems
Sample Answer
Problem Analysis

This problem requires finding the shortest distance between two nodes in a weighted, undirected graph. The most suitable algorithm for this task is Dijkstra's algorithm, which is specifically designed...

Approach
  1. Initialize Data Structures: Create a priority queue (min-heap) to store nodes and their cumulative distances from the starting node. Also, maintain a dictionary to track the shortest distances ...

Submit Your Answer
Markdown supported

Related Questions