Problem: 2050B - Transfusion
Language: java

Solution 1 (runtime: 639 ms):
/******************************************************************************

                            Online Java Compiler.
                Code, Compile, Run and Debug java program online.
Write your code in this editor and press "Run" button to execute it.

*******************************************************************************/
import java.util.*;

public class Main
{
	public static void main(String[] args) {
	    Scanner sc=new Scanner(System.in);
		int o=sc.nextInt();
		sc.nextLine();
// HashMap<Integer,Integer> map=new HashMap<>();

// int[] arr1=new int[o];
		for(int pp=0;pp<o;pp++){
		  int a =sc.nextInt();
		  //int b=sc.nextInt();
		  ////int c=sc.nextInt();
		  ////int d=sc.nextInt();
		  ////int e=sc.nextInt();
		  sc.nextLine();
		  int[] arr=new int[a];
		  long sum=0;
		  long sum1=0;
		  long sum2=0;
		  int odd=(a+1)/2;
		  int even=a/2;
		  for(int i=0;i<a;i++){
		      
		      int xx=sc.nextInt();
		      arr[i]=xx;
		      sum=sum+xx;
		      if(i%2==0){
		      sum1=sum1+xx;}
		      else{
		          sum2=sum2+xx;
		      }
		  }
		  long mod=sum/a;
		  if(sum%a!=0){
		      System.out.println("NO");
		      continue;
		  }
		  if(a==2){
		      if((arr[0]!=arr[1])){
		      System.out.println("NO");
		      continue;
		         }
		      else{
		      System.out.println("YES");
		      continue;
		       }
		  }
		
		  if(a==3){
		      if(arr[1]!=mod){
		      System.out.println("NO");
		      continue;
		      }
		  else{
		      System.out.println("YES");
		      continue;
		       }}
		  if(a>3){
		      if(sum1==(odd*mod)&&(sum2==(even*mod))){
		          System.out.println("YES");
		      continue;
		        }
		      else{
		          System.out.println("NO");
		          continue;
		  
		      }
		  }
		  
		}
		
		
		
}}


Solution 2 (runtime: 311 ms):
import java.io.BufferedOutputStream;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.PrintWriter;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.StringTokenizer;





public class Transfusion {

    private boolean debug = false;

    private PrintWriter outWriter;

    

    public static void main(String[] args) {

        new Transfusion().solve();

    }



    public void solve() {

        MyScanner sc = new MyScanner();

        outWriter = new PrintWriter(new BufferedOutputStream(System.out));

        int t = sc.nextInt();

        while (t-- > 0) {

            int n = sc.nextInt();

            int[] arr = new int[n];

            long sumOdd = 0;

            long sumEven = 0;

            for (int i = 0; i < n; i++) {

                arr[i] = sc.nextInt();

                if (i % 2 == 0) {

                    sumOdd += arr[i];

                } else {

                    sumEven += arr[i];

                }

            }

            int evenCount = n / 2;

            int oddCount = n - evenCount;

            if (sumEven % evenCount == 0 && sumOdd % oddCount == 0 && sumEven / evenCount == sumOdd / oddCount) {

                outWriter.println("YES");

            } else {

                outWriter.println("NO");

            }

        }

        outWriter.flush();

    }







    class PairIntegers implements Comparable<PairIntegers> {

        int left;

        int right;



        public PairIntegers(int x, int y) {

            this.left = x;

            this.right = y;

        }



        @Override

        public int compareTo(PairIntegers o) {

            int result = Integer.compare(left, o.left);

            if (result == 0) {

                result = Integer.compare(right, o.right);

            }

            return result;

        }



        @Override

        public boolean equals(Object obj) {

            if (this == obj)

                return true;

            if (obj == null || getClass() != obj.getClass())

                return false;

            PairIntegers other = (PairIntegers) obj;

            return left == other.left && right == other.right;

        }



        @Override

        public int hashCode() {

            return 31 * left + right;

        }



        @Override

        public String toString() {

            return "PairIntegers{" + "left=" + left + ", right=" + right + '}';

        }

    }



    public void debugPrintln(String str) {

        debugPrint(str.trim() + "\n");

    }



    public void debugPrint(String str) {

        if (debug) {

            outWriter.print(str);

        }

    }



    static final HashMap<Integer, Long> savedFactorials = new HashMap<>();



    public long ncr(int n, int r) {

        long mod = 998244353;

        savedFactorials.put(1, 1L);

        // f[n, r] = n! / {(n-r)! * r!}

        long res = 1;

        for (long x = n; x > n - r; x--) {

            res = (res * x) % mod;

        }

        long rFactorial = rFactorial(r);

        // debugPrintln(String.format("calculating (%d)C(%d): numerator: %d,

        // denominator: %d, result: %d", n, r, res, rFactorial, modDivision(res,

        // rFactorial, mod)));

        return modDivision(res, rFactorial, mod);

    }



    long rFactorial(int r) {

        if (savedFactorials.containsKey(r)) {

            return savedFactorials.get(r);

        }

        return r * rFactorial(r - 1);

    }



    long modDivision(long a, long b, long primeMod) {

        return a * modInverse(b, primeMod) % primeMod;

    }



    long modInverse(long b, long primeMod) {

        // b^(m-2) mod primeMod

        // using modular exponentation (divide & conquer)

        long result = 1;

        long base = b % primeMod; // Reduce base modulo m initially

        long e = primeMod - 2;

        while (e > 0) {

            if (e % 2 == 1) // If e is odd

                result = (result * base) % primeMod;

            base = (base * base) % primeMod; // Square the base

            e /= 2; // Divide exponent by 2

        }

        // debugPrintln(String.format("Modular inverse of %d is %d", b, result));

        return result;

    }



    class SieveOfErastothenes {

        List<Integer> primes;

        int limit;

        int[] arr;

        Map<Integer, List<Integer>> factors;



        public SieveOfErastothenes(int limit) {

            this.arr = new int[limit + 1];

            this.limit = limit;

            this.factors = new HashMap<>();

        }



        public Map<Integer, List<Integer>> getFactors() {

            for (int i = 2; i <= limit; i++) {

                for (int j = 1; i * j <= limit; j++) {

                    factors.putIfAbsent(i * j, new ArrayList<>());

                    factors.get(i * j).add(i);

                }

            }

            return factors;

        }



        public List<Integer> getPrimes() {

            List<Integer> result = new ArrayList<>();

            int i;

            for (i = 2; i <= Math.sqrt(limit); i++) {

                if (arr[i] == 0) {

                    result.add(i);

                    for (int j = 1; j * i <= limit; j++) {

                        arr[j * i] = -1; // j * i == xyz

                    }

                }

            }

            while (i <= limit) {

                if (arr[i] == 0)

                    result.add(i);

                i++;

            }

            return result;

        }

    }



    class DisJointSetUnion {

        int[] arr;



        public DisJointSetUnion(int size) {

            size++;

            this.arr = new int[size];

            for (int i = 0; i < size; i++) {

                arr[i] = i;

            }

        }



        public Map<Integer, Set<Integer>> expandSets() {

            Map<Integer, Set<Integer>> result = new HashMap<>();

            for (int i = 1; i < arr.length; i++) {

                int parent = find(i);

                if (!result.containsKey(parent)) {

                    result.put(parent, new HashSet<>());

                }

                result.get(parent).add(i);

                result.put(i, result.get(parent));

            }

            return result;

        }



        public int find(int i) {

            if (arr[i] == i) {

                return i;

            }

            arr[i] = find(arr[i]);

            return arr[i];

        }



        public int union(int i, int j) {

            int iKaBaap = find(i);

            int jKaBaap = find(j);

            if (iKaBaap > jKaBaap) {

                arr[iKaBaap] = jKaBaap;

            } else {

                arr[jKaBaap] = iKaBaap;

            }

            return jKaBaap;

        }

    }



    // -----------PrintWriter for faster output---------------------------------

    public static PrintWriter out;



    // -----------MyScanner class for faster input----------

    public static class MyScanner {

        BufferedReader br;

        StringTokenizer st;



        public MyScanner() {

            br = new BufferedReader(new InputStreamReader(System.in));

        }



        String next() {

            while (st == null || !st.hasMoreElements()) {

                try {

                    st = new StringTokenizer(br.readLine());

                } catch (IOException e) {

                    e.printStackTrace();

                }

            }

            return st.nextToken();

        }



        int nextInt() {

            return Integer.parseInt(next());

        }



        long nextLong() {

            return Long.parseLong(next());

        }



        double nextDouble() {

            return Double.parseDouble(next());

        }



        String nextLine() {

            String str = "";

            try {

                str = br.readLine();

            } catch (IOException e) {

                e.printStackTrace();

            }

            return str;

        }



    }

    // --------------------------------------------------------

}