Getty

Diddy should serve the remainder of his 50-month sentence in a federal prison as close to the New York metropolitan area as possible ... at least according to the judge who imposed the sentence.

According to new legal docs, obtained by TMZ, Judge Arun Subramanian stops short of naming a specific prison where Diddy should do hard time and instead says Diddy should stay close to home.

Besides geography, the judge says Diddy should be shipped off to a prison with "any available substance abuse program, including the Bureau of Prisons' Residential Drug Abuse Program (RDAP), for which he qualifies." Federal Bureau of Prisons

We broke the story ... Diddy's legal team says FCI Fort Dix in New Jersey is the best landing spot for Diddy because there he can "address drug

See Full Page