31.下一个排列

next-permutation

题目描述

实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。

必须原地修改，只允许使用额外常数空间。

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

代码

package pid31;

public class Solution {
	public void nextPermutation(int[] nums){
		int first = 0;
		int second = 0;
		for(int i=nums.length-2;i>=0;i--){
			if(nums[i]<nums[i+1]){
				first = i;
				break;
			}
		}
		for(int i=nums.length-1;i>=0;i--){
			if(nums[i]>nums[first]){
				second = i;
				break;
			}
		}
		if(first == 0 && second == 0){
			// nums是一个递减数组
			reverse(nums,0);
			return;
		}
		swap(nums,first,second);
		reverse(nums,first+1);
	}
	
	public void swap(int[] nums,int i,int j){
		int temp = nums[i];
		nums[i] = nums[j];
		nums[j] = temp;
	}
	
	public void reverse(int[] nums,int start){
		int i = start;
		int j = nums.length-1;
		while(i<j){
			int temp = nums[i];
			nums[i] = nums[j];
			nums[j] = temp;
			i++;
			j--;
		}
	}
	
	public static void main(String[] args){
		Solution sol = new Solution();
		int[] nums = {1,2,3};
		sol.nextPermutation(nums);
		for(int curr:nums){
			System.out.print(curr+" ");
		}
	}
}

性能表现