$(document).ready(function() {
  $('.menu > ul > li > ul').hide(); /* close everything */
  $('.menu > ul > li > a.selected + ul').show(); /* open, if root */
  $('.menu > ul > li > ul:has(.selected)').show(); /* open, if child */
});<!-- 0.0915s -->