fix(tree): use in-memory data first to query total difficulty

fix(tree): use in-memory data first to query total difficulty #34971