Codeforces Round 918 (Div. 4) E题

题目+链接：Codeforces Round 918 (Div. 4) E. Romantic Glasses java

传送门：https://codeforces.com/contest/1915/problem/E

分析：

首先，喝酒顺序不重要，只要保证一段子序列中男女喝酒一样多。
设男生喝酒为正数，女生喝酒为负数。
我们可以构造一个式子：a1-a2+a3-a4+…-an
记录一下构造式子前缀和/差 -> sum，然后通过set去维护。如果set中存在sum，说明之前有一个相等的前缀和/差sum，即之前sum与该sum这段数相加为0，找到答案；否则将sum加入到set中。

最开始没喝酒时为0也要加入要set中
a[i]的数量级为10⁹，定义sum和set类型为long

代码：

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;

public class Main {
    public static void main(String[] args) {
    	Scanner sc = new Scanner(System.in);
    	int t = sc.nextInt();
    	h:while(t-->0) {
    		int n = sc.nextInt();
            // 求前缀和/差
    		long sum = 0;
    		// 记录前缀和/差
    		Set<Long> set = new HashSet<>();
    		set.add((long) 0);
    		// 是否有连续子阵列
    		boolean flag = false;
    		for(int i = 0;i < n;i++) {
    			int tmp = sc.nextInt();
    			if(i%2==1) tmp *= -1;
    			sum += tmp;
    			if(set.contains(sum)) flag = true;
    			set.add(sum);
    		}
    		if(flag==true) System.out.println("YES");
    		else System.out.println("NO");
    		set.clear();
    	}
    	sc.close();
    }
}