尺取法:PIPI的目标Ⅲ

本文介绍了如何使用尺取法优化求解PIPI问题,通过排序和迭代策略,避免了O(n^3)的时间复杂度,提高了算法效率。关键步骤包括枚举a,动态调整b和c,利用HashSet去重并重写equals和hashCode方法。
摘要由CSDN通过智能技术生成

尺取法:PIPI的目标Ⅲ

问题:

在这里插入图片描述
在这里插入图片描述

思路:

  首先对数组R进行排序,如果直接枚举a,b,c的话,时间复杂度为O(n^3),肯定会超时,那么我们接下来考虑使用尺取法来解决此题。
  我们只枚举a,然后令b=a+1,即b为a的下一位,c=n。枚举过程如下:
  如果此时R[a]+R[b]+R[c]==0,那么满足要求,我们输出结果,然后令b++,c- -,之后继续进行和的判断。
  如果此时R[a]+R[b]+R[c]<0,那么显然我们的值小了。由于我们枚举的是a,所以此时a不能动,那么只能动b或者c,而只有动b才能让它变大,所以让b++,之后继续进行和的判断。
  如果此时R[a]+R[b]+R[c]>0,那么显然我们的值大了。a不能动,那么只能动b或者c,而只有动c才能让它变小,所以让c- -,之后继续进行和的判断。
  当b在c之后或b和c重叠时,则我们已经枚举完了当前a的所有情况,则a移到下一位,继续进行枚举判断过程

  题目要求我们输出不重复的答案,因此我们将< a, b, c >定义为三元组TriGroup,用一个HashSet来进行去重。需要注意的是,我们必须重写TriGroup的equals方法,因为我们这里的逻辑是当三元组中a,b,c的值对应相等,则认为两个TriGroup是同一个,而默认的equals方法是通过==判断实现的。hashCode方法也要重写,因为HashSet的add方法,调用的实际上是其内置的HashMap的put方法,该方法是要使用到键的hash值的,而计算hash值就是用的hashCode方法,因此我们重写hashCode方法,使其返回a,b,c共同决定的hash值

代码:

import java.util.*;

public class Main {
    static int[] array = new int[1005];
    static Set<TriGroup> set = new HashSet<>();
    public static void main(String[] args) {
        int a, b, c, n, i, aIndex, bIndex, cIndex;
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNextInt()) {
            set.clear();
            n = scanner.nextInt();
            for (i = 0; i < n; i++) {
                array[i] = scanner.nextInt();
            }
            Arrays.sort(array, 0, n);
            for (i = 0; i < n - 2; i++) {
                aIndex = i;
                bIndex = i + 1;
                cIndex = n - 1;
                a = array[aIndex];
                b = array[bIndex];
                c = array[cIndex];
                while (bIndex < cIndex) {
                    if (a + b + c < 0) {
                        bIndex++;
                        b = array[bIndex];
                    }
                    if (a + b + c > 0) {
                        cIndex--;
                        c = array[cIndex];
                    }
                    if (a + b + c == 0 && cIndex > bIndex) {
                        TriGroup triGroup = new TriGroup(a, b, c);
                        if (!set.contains(triGroup)) {
                            set.add(triGroup);
                            System.out.println(a + " " + b + " " + c);
                        }
                        bIndex++;
                        b = array[bIndex];
                        cIndex--;
                        c = array[cIndex];
                    }
                }
            }
        }
    }


}

class TriGroup {
    public int a;
    public int b;
    public int c;

    public TriGroup(int a, int b, int c) {
        this.a = a;
        this.b = b;
        this.c = c;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        TriGroup triGroup = (TriGroup) o;
        return a == triGroup.a && b == triGroup.b && c == triGroup.c;
    }

    @Override
    public int hashCode() {
        return Objects.hash(a, b, c);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

happy19991001

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值