TypeScript: Map certain types inside function argument types w/out intermediate type parameters

Question

For example, this works as I expected: Playground #1

const f = (
  arg: {
    [K in keyof A]: A[K]
  }
) =>