码迷,mamicode.com
首页 > 其他好文 > 详细

【HackerRank】Sherlock and Array

时间:2014-08-02 12:36:13      阅读:185      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   java   strong   io   for   ar   

Watson gives an array A1,A2...AN to Sherlock. Then he asks him to find if there exists an element in the array, such that, the sum of elements on its left is equal to the sum of elements on its right. If there are no elements to left/right, then sum is considered to be zero.
Formally, find an i, such that, A1+A2...Ai-1 = Ai+1+Ai+2...AN.

Input Format
The first line contains T, the number of test cases. For each test case, the first line contains N, the number of elements in the array A. The second line for each testcase contains N space separated integers, denoting the array A.

Output Format
For each test case, print YES if there exists an element in the array, such that, the sum of elements on its left is equal to the sum of elements on its right, else print NO.

Constraints
1 ≤ T ≤ 10
1 ≤ N ≤ 105
1 ≤ Ai ≤ 2*104 for 1 ≤ i ≤ N


先求出数组总的sum,然后遍历一遍数组,遍历的过程中分别累加(减)求出一个元素的左边元素之和left(left的累加在这个元素上一个元素的循环中完成,见代码第30行)和右边元素之和right,然后比较是否相等即可,时间复杂度O(n),代码如下:

 1 import java.io.*;
 2 import java.util.*;
 3 import java.text.*;
 4 import java.math.*;
 5 import java.util.regex.*;
 6 
 7 public class Solution {
 8     public static void main(String[] args) {
 9         Scanner in = new Scanner(System.in);
10         int t = in.nextInt();
11         while(t-- > 0){
12             int n = in.nextInt();
13             int[] num = new int[n];
14             int sum = 0;
15             
16             for(int i = 0;i < n;i++){
17                 num[i] = in.nextInt();
18                 sum += num[i];
19             }
20             int left = 0;
21             int right = sum;
22             boolean find = false;
23             for(int i = 0;i < n;i++){
24                 right -= num[i];
25                 if(left == right)
26                 {
27                     find = true;
28                     break;
29                 }
30                 left += num[i];
31             }
32             if(find)
33                 System.out.println("YES");
34             else
35                 System.out.println("NO");
36         }
37       }
38 }

 

【HackerRank】Sherlock and Array,布布扣,bubuko.com

【HackerRank】Sherlock and Array

标签:style   blog   color   java   strong   io   for   ar   

原文地址:http://www.cnblogs.com/sunshineatnoon/p/3886696.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!