Use case for useLayoutEffect + useState vs useMemo?

Question

React

I saw this answer: useMemo vs. useEffect + useState and it fails useEffect well, but in my case I want to perform an expensive operation that changes the DOM as early as possible. Would useMemo() still be recommended instead of useLayoutEffect() for state updates? Does double effect rendering -> state update reflect any performance gain?
useLayoutEffect() script:

useLayoutEffect(() => {
    const tokens = expensiveOperationGeneratingClasses(param1)
    setTokens(tokens)
}, 
[param1])

 render (
  <>
   {
       tokens.map(token => <span className={token.id}/>)
   }
  </>
 )

useMemo script:

const tokens = useMemo(() => {
     return expensiveOperationGeneratingClasses(param1)
},
[param1]

 render (
  <>
   {
       tokens.map(token => <span className={token.id}/>)
   }
  </>
 )

Actually I realized that I'm not doing DOM operations but just generating class names before rendering tags to avoid flickering, so I think I'm better off using useMemo, am I right?