In this HackerRank Fun Game problem solution we have given the value of two N-sized integer arrays, A and B., and is played by two players P1 and P2. can you determine the outcome of the game? Print First if P1 will win, Second if P2 will win, or Tie if they will tie. Assume both players always move optimally.
Problem solution in Python.
import sys # Process input data = sys.stdin.readlines() T = int(data[0]) def run_game(N, a, b): c = [a[i] + b[i] for i in range(N)] # a-b cs = [i for i, x in sorted( enumerate(c), key=(lambda y : y[1]))] # keys that sort c # Play game p1, p2 = (0, 0) # payoffs to the two players for i in range(N): if (i % 2)==0: # P1's turn p1 += a[cs.pop(-1)] else: # P2's turn p2 += b[cs.pop(-1)] # Print results if p1 > p2: print("First") elif p1==p2: print("Tie") else: print("Second") for k in range(T): N = int(data[3*k + 1]) a = list(map(int, data[3*k + 2].split())) b = list(map(int, data[3*k + 3].split())) run_game(N, a, b)
Problem solution in Java.
import java.io.*; import java.util.*; public class Solution { public static void main(String[] args) { Scanner input = new Scanner(System.in); int T = input.nextInt(); for(int t=0; t<T; t++){ int n = input.nextInt(); int[] a = new int[n]; int[] b = new int[n]; int aScore = 0, bScore = 0; TreeSet<Pair> c = new TreeSet<>(); for(int i=0; i<n; i++){ a[i] = input.nextInt(); } for(int i=0; i<n; i++){ b[i] = input.nextInt(); c.add(new Pair(a[i]+b[i], i)); } int turn = 0; for(Pair p:c){ if(turn == 0) aScore += a[p.index]; else bScore += b[p.index]; turn = 1 - turn; } if(aScore > bScore){ System.out.println("First"); }else if(aScore == bScore){ System.out.println("Tie"); }else{ System.out.println("Second"); } } } public static class Pair implements Comparable<Pair>{ int sum; int index; Pair(int s, int i){ this.sum = s; this.index = i; } @Override public int compareTo(Pair p){ if(this.sum < p.sum){ return 1; }else{ return -1; } } public String toString(){ return "{ s = "+sum+", i = "+index+" }"; } } }
Problem solution in C++.
#include <cmath> #include <cstdio> #include <vector> #include <iostream> #include <algorithm> using namespace std; const int Maxn = 1005; int t; int n; int sum; int a[Maxn]; int main() { scanf("%d", &t); while (t--) { scanf("%d", &n); sum = 0; for (int i = 0; i < n; i++) { scanf("%d", &a[i]); sum += a[i]; } for (int i = 0; i < n; i++) { int tmp; scanf("%d", &tmp); a[i] += tmp; sum -= tmp; } sort(a, a + n); for (int i = n - 1; i >= 0; i--) if ((n - 1 - i) % 2 == 0) sum += a[i]; else sum -= a[i]; if (sum > 0) printf("Firstn"); else if (sum == 0) printf("Tien"); else printf("Secondn"); } return 0; }
Problem solution in C.
#include<stdio.h> int solve(int *a, int *b, int n, int max) { int temp = 0, loss, ret = a[0], i; loss = b[0] - ( max - a[0] ); for( i = 1 ; i < n ; i++ ) { if( ( b[i] - ( max - a[i] ) ) > loss ) { loss = b[i] - ( max - a[i] ); ret = a[i]; temp = i; } } a[temp] = b[temp] = 0; return ret; } int main() { int t, n, start, i; long P1, P2; scanf("%d", &t); while(t--) { scanf("%d", &n); int a[n], b[n], max = -1; for( i = 0 ; i < n ; i++ ) { scanf("%d", a + i); if( max < a[i] ) { max = a[i]; } } for( i = 0 ; i < n ; i++ ) { scanf("%d", b + i); } start = 1, P1 = P2 = 0; for( i = 0 ; i < n ; i++ ) { start > 0 ? ( P1 += solve(a, b, n, max) ): ( P2 += solve(b, a, n, max) ); start = !start; } puts(P1 > P2 ? "First" : P1 < P2 ? "Second" : "Tie"); } return 0; }