The rise of One Nation is partly behind the government's decision to take on tax reform in this year's budget, the prime minister has admitted.