Intersection of Two Linked Lists

 1 /**
 2  * Definition for singly-linked list.
 3  * struct ListNode {
 4  *     int val;
 5  *     ListNode *next;
 6  *     ListNode(int x) : val(x), next(NULL) {}
 7  * };
 8  */
 9 class Solution {
10 public:
11     ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
12         if(!headA || !headB) return NULL;
13         
14         if(getLength(headA) > getLength(headB)){
15             int len = getLength(headA);
16             while(len > getLength(headB)){
17                 headA = headA->next;
18                 len--;
19             }
20         }
21         else{
22             int len = getLength(headB);
23             while(len > getLength(headA)){
24                 headB = headB->next;
25                 len--;
26             }
27         }
28         
29         while(headA){
30             if(headA == headB)
31                 return headA;
32             headA = headA->next;
33             headB = headB->next;
34         }
35         return NULL;
36     }
37     
38     int getLength(ListNode* head){
39         int result = 0;
40         
41         while(head){
42             head = head->next;
43             result++;
44         }
45         return result;
46     }
47 };