宣告一個陣列代表每一個人的最底層朋友是誰，每一個人預設值為 0。如果收到人的時候這個人的陣列值為 0，代表還沒有紀錄，所以將陣列值設為自己。

宣告一個函式 findEnd 用來尋找最底層的朋友，有一個參數 N，如果 N 的陣列值為 0 或是自己就 return N，否則就 return findEnd(陣列值[N])。

收到資料的時候將 陣列值[findEnd(B)] 設為 findEnd(A)。最後只需要確認 陣列值[P] 是否等於 陣列值[Q]，如果相同就代表這兩人是朋友。

範例程式碼