Fair Election (FAIRELCT) Solution -Codechef JanuaryLong Challenge
Problem Statement
Elections are coming soon. This year, two candidates passed to the final stage. One candidate is John Jackson and his opponent is Jack Johnson.
During the elections, everyone can vote for their favourite candidate, but no one can vote for both candidates. Then, packs of votes which went to the same candidate are formed. You know that for John Jackson, there are N packs containing A1,A2,…,AN votes, and for Jack Johnson, there are M packs containing B1,B2,…,BM votes.
The winner is the candidate that has strictly more votes than the other candidate; if both have the same number of votes, there is no winner. You are a friend of John Jackson and you want to help him win. To do that, you may perform the following operation any number of times (including zero): choose two packs of votes that currently belong to different candidates and swap them, i.e. change the votes in each of these packs so that they would go to the other candidate.
You are very careful, so you want to perform as few swaps as possible. Find the smallest number of operations you need to perform or determine that it is impossible (should have been written possible) to make John Jackson win.
Input
- The first line of the input contains a single integer T denoting the number of test cases. The description of T test cases follows.
- The first line of each test case contains two space-separated integers N and M.
- The second line contains N space-separated integers A1,A2,…,AN.
- The third line contains M space-separated integers B1,B2,…,BM.
Output
For each test case, print a single line containing one integer ― the smallest number of swaps needed to make John Jackson win, or −1 if it is impossible.
Constraints
- 1≤T≤103
- 1≤N,M≤103
- 1≤Ai≤106 for each valid i
- 1≤Bi≤106 for each valid i
- the sum of N over all test cases does not exceed 104
- the sum of M over all test cases does not exceed 104
Subtasks
Subtask #1 (20 points):
- A1=A2=…=AN
- B1=B2=…=BM
Subtask #2 (80 points): original constraints
Example Input
2
2 3
2 2
5 5 5
4 3
1 3 2 4
6 7 8
Example Output
2
1
Explanation
Example case 1: We can perform two swaps ― each time, we swap a pack of 2 votes from A and a pack of 5 votes from B. After that, John Jackson gets 5+5=10 votes and Jack Johnson gets 2+2+5=9 votes.
Example case 2: We can swap the pack of 1 vote from A and the pack of 8 votes from BB. After that, John Jackson gets 8+3+2+4=17 votes and Jack Johnson gets 6+7+1=14 votes.
Code:
import java.util.*;
import java.lang.*;
import java.io.*;
import java.util.Collections;
class Codechef
{
public static void main (String[] args) throws java.lang.Exception
{
// your code goes here
Scanner sc = new Scanner(System.in);
int t = sc.nextInt();
while(t-- > 0){
int n = sc.nextInt();
int m = sc.nextInt();
int ar1[] = new int[n];
int ar2[] = new int[m];
int count =0; int sum1 =0;
int sum2 =0;
for(int i=0;i<n;i++){
ar1[i]=sc.nextInt();
sum1 += ar1[i];
}
for(int i=0;i<m;i++){
ar2[i]=sc.nextInt();
sum2 += ar2[i];
}
if(sum1 > sum2){ //if no work required
System.out.println("0");
continue;
}
Arrays.sort(ar1);
Arrays.sort(ar2);
for(int i=0; i<m/2; ++i) {
int temp = ar2[i];
ar2[i] = ar2[m-1-i];
ar2[m-1-i] = temp;
}
int i=0,j=0;
while(i< n && j< m){
sum1 += ar2[j]-ar1[i];
sum2 += ar1[i]-ar2[j];
if(sum1 > sum2){
break;
}
i++;j++;
count++;
}
if(sum1 <= sum2 ){
System.out.println("-1");
continue;
}
System.out.println(count+1);
}
}
}
Hope you would have liked the article. Please give 50 clap to this article and follow me for more future programming related blogs.
References