Realmente esto no es cuestión de nivel @JUMASOL, simplente viendo el manual y viendo que te extrae el path, puedes hacer la comparación con la raiz y hacer tu condicional, no es cuestión de buscar por internet y pegar códigos sin saber lo que hacen.
Si lo que esperas viniendo al foro es que la gente te de soluciones directas sin tú querer aprender pues creo vas a tener un mal momento, la idea es que encuentres las soluciones por ti mismo usando las herramientas que te podamos sugerir.
Te invito que a la siguiente trates de reflexionar un poco el problema y trates de encontrar un algoritmo que se ajuste a lo que necesitas y en base a eso te podemos apoyar
Saludos.