枚举算法的思想例题2

枚举算法的思想例题

solution1：小Hi和小Ho的礼物

hihoCoder #1505题：小Hi和小Ho的礼物 http://hihocoder.com/problemset/problem/1505

 static void solution2(){
                int N=in.nextInt();
                int [] re=new int[N+1];
                int [] cou=new int[N*2];
                for(int i=1;i<N;i++){
                    re[i]=in.nextInt();
                    cou[re[i]]++;
                }
                Map<Integer,Integer> map=new HashMap<>();
                int ans=0;
                for(int i=1;i<=N;i++){
                    for(int j=i+1;j<=N;j++){
                        int x=re[i]+re[j];
                        Integer ival = map.get(x);
                        if(ival==null) ival=0;
                        map.put(x,++ival);
                    }
                }
                for(int i=1;i<=N;i++){
                    for(int j=i+1;j<=N;j++){
                        if(re[i]!=re[j]){
                            Integer val = map.get(re[i] + re[j]);
                            ans+= val-(cou[re[i]]+cou[re[j]]-2)-1;
                        }else{
                            Integer val = map.get(re[i] + re[j]);
                            ans+= val-(cou[re[i]]+cou[re[j]]-4)-1;
                        }
                    }
                }
                System.out.println(ans);
    }

solution2：互补二元组

　

 解法

static void solution3(){
        int N=in.nextInt();
        Map<Integer,Integer> map=new HashMap<>();
        int ans=0;
        for(int i=0;i<N;i++)
        {
            int a,b;
            a=in.nextInt();
            b=in.nextInt();
            int temp=a-b;
            if (map.containsKey(-temp)){
                ans+=map.get(-temp);
            }else{
                Integer integer = map.get(temp);
                if (integer==null)  integer=0;
                map.put(temp,++integer);
            }
        }
        System.out.println(ans);

    }

　

双指针 

解决以下代码：引入第二个下标J来降低复杂度

for(int i = 0;i < n;i++)
{
    for(int j = i + 1;j < n;j++)
    {
        ...
    }
}

 https://leetcode-cn.com/tag/two-pointers/

 

例1

 给定N个整数A₁，A₂，...，A_N，以及一个正整数k。问在所有的大于等于k的两个数的差（A_i-A_j）中，最小的差是多少？（N<=100000）

思路：

 首先对A数组排序，比如假设排好序的A数组是：A=[1,3,7,8,10,15]，k=3，这时我们枚举两个数中较小的是A[i]，较大的是A[j]；对A[i]来说，我们要找到最优的A[j]，也就是最小的A[j]满足A[j] - A[i] >= k

 

 - code

 static void solution(){
        int N,k;
        N=in.nextInt();
        k=in.nextInt();
        int [] a=new int[N*2];
        for(int i=0;i<N;i++){
            a[i]=in.nextInt();
        }
        Arrays.sort(a,0,N);
        if(a[N-1]-a[0]<k) {
            throw new ArrayIndexOutOfBoundsException();
        }
        int ans=a[N-1]-a[0];
        for(int i=0,j=0;i<N;i++){//1 3 7 8 10 15
            while(j < N && a[j] - a[i] < k)
                j++;
            if(a[j] - a[i] >= k && a[j] - a[i] < ans)
                ans = a[j] - a[i];


        }
        System.out.println(ans);
    }