【回溯】AcWing4498. PointerToThe-指针

🔗原题

给定一个360°的全圆量角器。
初始时，量角器上的指针指向刻度 0。
现在，请你对指针进行 n 次拨动操作，每次操作给定一个拨动角度 ai，由你将指针拨动 ai 度，每次的拨动方向（顺时针或逆时针）由你自由决定。
请你判断，能否通过合理选择每次拨动的方向，使得指针最终仍然指向刻度 0。

输入格式
第一行包含整数 n。
接下来 n 行，每行包含一个整数 ai，表示一次操作的拨动角度。
输出格式
如果可以做到指针最终仍然指向刻度 0，则输出 YES，否则输出 NO。

数据范围
前 4 个测试点满足 1≤n≤3。
所有测试点满足 1≤n≤15，1≤ai≤180。

输入样例1：
3
10
20
30
输出样例1：
YES

输入样例2：
3
10
10
10
输出样例2：
NO

输入样例3：
3
120
120
120
输出样例3：
YES

解题思路

回溯法
通过观察数据范围，可以使用回溯法
以下几种情况停止递归:
1，已经找到有符合的情况
2，最后一个角度的情况已经处理

在最后一个角度的情况已经处理并且对360取余为0的时，有我们需要的答案，进行标记

每次可以选择增加角度或减少角度

代码

import java.util.Scanner;

public class Main {
    static int falg = 0;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int nums[] = new int[n];
        for(int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
        }
        help(nums, 0, 0);
        System.out.println(falg==1?"YES":"NO");
    }
    public static void help(int nums[], int p, int sum) {
        if(p == nums.length && sum%360==0) {
            falg = 1;
            return;
        }
        if(falg == 1 || p >= nums.length) return;
        help(nums, p + 1, sum + nums[p]);
        help(nums, p + 1, sum - nums[p]);
    }
}