MT2048-三连Java版（前缀和解法）

洒水水儿

已于 2023-12-03 20:57:00 修改

阅读量54

点赞数

文章标签： java 算法开发语言

于 2023-12-03 20:54:44 首次发布

本文链接：https://blog.csdn.net/cangshanjiang/article/details/134770265

版权

该代码片段是一个Java程序，用于在一个整数数组中找到所有的三分之一点（即其和为数组总和的三分之一），并在找到三分之二点之前记录这些点的数量。程序考虑了边界条件和优化计算过程。

摘要由CSDN通过智能技术生成

import java.util.Scanner;
import java.util.*;

class Main {

   public static void main(String[] args) {
      final int  N = 100010;
      Scanner input = new Scanner(System.in);
      // code here
      int n = input.nextInt();
      int[] arr = new int[N];
      int sum = 0, s = 0, ans = 0, count = 0;

      for (int i = 1; i <= n; i++ ) {
         arr[i] = input.nextInt();
         sum += arr[i];
      }
      for (int i = 1; i <= n; i++ ) {
         s += arr[i];
         if (s*3 == sum * 2 && i > 1 && i < n) {
            ans += count;
         }
         if (s*3 == sum){
            count++;
         }
      }
      System.out.println(ans);
      input.close();
   }
}

拆解为找三分之一点和三分之二点，在找到三分之二点之前记录所有三分之一点的数目，遇到三分之二点后进行结算

注意边界值，并且需要先判断三分之二点