Fact: A/J actually curated the piano part in the beginning of Puzzle Pieces when he was a kid. The little line just stuck with him up until he actually used it in the song.
Bonus fact: if you want to know how to play it, I took the liberty of figuring it out for y'all.