[Codeforces] 1300A: Non-zero

https://codeforces.com/problemset/problem/1300/A

Problem - 1300A - Codeforces

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		
		int t = Integer.parseInt(bf.readLine());
		for(int tc=0; tc<t; tc++) {
			int n = Integer.parseInt(bf.readLine());
			int[] arr = new int[n];
			StringTokenizer st = new StringTokenizer(bf.readLine());
			for(int i=0; i<n; i++)
				arr[i] = Integer.parseInt(st.nextToken());
			
			int ans = 0;	// 최소 단계
			int sum = 0;	// 배열 모든요소의 합
			
			// 배열의 요소에 0이 존재하면 곱이 0이되므로 0을 1로만든다.
			for(int i=0; i<n; i++) {
				if(arr[i] == 0) {
					arr[i] ++;
					ans ++;
				}
			}
			
			// 배열 모든 요소의 합
			for(int i=0; i<n; i++)
				sum += arr[i];
			
			// 배열 요소의 합이 0이되면 안됨,
			ans += (sum == 0) ? 1 : 0;
			
			System.out.println(ans);
				
		}
		
		bf.close();
	}

}

풀이

배열의 모든 요소들의 합과 곱이 0이 되지 않도록 만들어야 한다.

 1) 먼저 배열의 요소에 0이 있는경우 -> 전부 1로 만들어준다.

 2) 배열의 모든 요소의 합이 0인경우 -> +1을 해준다.(합이 음수이든 양수이든 상관없이 0만 아니면 됨.)