java中多重循环的简化

最新推荐文章于 2023-04-30 16:32:36 发布

原创最新推荐文章于 2023-04-30 16:32:36 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

32 篇文章

订阅专栏

要求：4个循环，是否存在和为m的情况。

最暴力的方法，四个for:

public class Forsearch {
	public static void main(String[] args){
		int m = 100;
		int d;
		int[] array = new int[1000];
		//a+b+c+d = m;          d = m -a -b -c;
		double begin = System.currentTimeMillis();
		for(int a=1;a<=1000;a++){
			for(int b=1;b<=1000;b++){
				for(int c=1;c<=1000;c++){
					for(int d=1;d>1000;d++){
                                            if(a+b+c+d==m){
                                                xxxx;
                                            }
                                        }
				}
			}
		}
		double end = System.currentTimeMillis();
		System.out.println(end-begin);
		
	}
}

由于m已给出，可将其中一个未知的d,进行二分查找。

即：d = m-a-b-c;

此时只需要 O(n3) * O(log n)

import java.util.Arrays;


public class Forsearch {
	public static void main(String[] args){
		int m = 4;
		int d;
		int[] array = new int[1000];
		for(int i=0;i<1000;i++){
			array[i]=i+1;
		}
		//a+b+c+d = m;          d = m -a -b -c;
		double begin = System.currentTimeMillis();
		for(int a=1;a<=1000;a++){
			for(int b=1;b<=1000;b++){
				for(int c=1;c<=1000;c++){
					if(Arrays.binarySearch(array, m-a-b-c)!=-1){
						System.out.println(true);
					}
				}
			}
		}
		double end = System.currentTimeMillis();
		System.out.println(end-begin);
		
	}
}

但是这样还是太慢，考虑两个变量都进行二叉查找。但程序中不允许两个求知变量。

所以可以考虑用b,c之和装进一个整个进行查找。

同理，3个4个都可以这样做。

即在4-4000中查找m.

import java.util.Arrays;

public class Forsearch {
	public static void main(String[] args) {
		int m = 4;
		int d;
		int[] array = new int[3996];
		for (int i = 0; i < 3996; i++) {
			array[i] = i + 4;
		}
		if (Arrays.binarySearch(array, m) != -1) {
			System.out.println(true);
		}
		double end = System.currentTimeMillis();

	}
}