Codeforcess solution 1178-A A. Prime Minister C++, Java, Js and Python

Algorithm

Alice is the leader of the State Refactoring Party, and she is about to become the prime minister.

The elections have just taken place. There are $n"> n$ parties, numbered from $1"> 1$ to $n"> n$ . The $i"> i$ -th party has received $ai"> a i$ seats in the parliament.

Alice's party has number $1"> 1$ . In order to become the prime minister, she needs to build a coalition, consisting of her party and possibly some other parties. There are two conditions she needs to fulfil:

The total number of seats of all parties in the coalition must be a strict majority of all the seats, i.e. it must have strictly more than half of the seats. For example, if the parliament has $200"> 200$ (or $201"> 201$ ) seats, then the majority is $101"> 101$ or more seats.
Alice's party must have at least $2"> 2$ times more seats than any other party in the coalition. For example, to invite a party with $50"> 50$ seats, Alice's party must have at least $100"> 100$ seats.

For example, if $n=4"> n = 4$ and $a=[51,25,99,25]"> a = [51, 25, 99, 25]$ (note that Alice'a party has $51"> 51$ seats), then the following set $[a1=51,a2=25,a4=25]"> [a 1 = 51, a 2 = 25, a 4 = 25]$ can create a coalition since both conditions will be satisfied. However, the following sets will not create a coalition:

$[a2=25,a3=99,a4=25]"> [a 2 = 25, a 3 = 99, a 4 = 25]$ since Alice's party is not there;
$[a1=51,a2=25]"> [a 1 = 51, a 2 = 25]$ since coalition should have a strict majority;
$[a1=51,a2=25,a3=99]"> [a 1 = 51, a 2 = 25, a 3 = 99]$ since Alice's party should have at least $2"> 2$ times more seats than any other party in the coalition.

Alice does not have to minimise the number of parties in a coalition. If she wants, she can invite as many parties as she wants (as long as the conditions are satisfied). If Alice's party has enough people to create a coalition on her own, she can invite no parties.

Note that Alice can either invite a party as a whole or not at all. It is not possible to invite only some of the deputies (seats) from another party. In other words, if Alice invites a party, she invites all its deputies.

Find and print any suitable coalition.

Input

The first line contains a single integer $n"> n$ ( $2\leqn\leq100"> 2 \leq n \leq 100$ ) — the number of parties.

The second line contains $n"> n$ space separated integers $a1,a2,\dots,an"> a 1, a 2, \dots, a n$ ( $1\leqai\leq100"> 1 \leq a i \leq 100$ ) — the number of seats the $i"> i$ -th party has.

Output

If no coalition satisfying both conditions is possible, output a single line with an integer $0"> 0$ .

Otherwise, suppose there are $k"> k$ ( $1\leqk\leqn"> 1 \leq k \leq n$ ) parties in the coalition (Alice does not have to minimise the number of parties in a coalition), and their indices are $c1,c2,\dots,ck"> c 1, c 2, \dots, c k$ ( $1\leqci\leqn"> 1 \leq c i \leq n$ ). Output two lines, first containing the integer $k"> k$ , and the second the space-separated indices $c1,c2,\dots,ck"> c 1, c 2, \dots, c k$ .

You may print the parties in any order. Alice's party (number $1"> 1$ ) must be on that list. If there are multiple solutions, you may print any of them.

Examples
input
Copy
3
100 50 50
output
Copy
2
1 2
input
Copy
3
80 60 60
output
Copy
0
input
Copy
2
6 5
output
Copy
1
1
input
Copy
4
51 25 99 25
output
Copy
3
1 2 4

Note

In the first example, Alice picks the second party. Note that she can also pick the third party or both of them. However, she cannot become prime minister without any of them, because $100"> 100$ is not a strict majority out of $200"> 200$ .

In the second example, there is no way of building a majority, as both other parties are too large to become a coalition partner.

In the third example, Alice already has the majority.

The fourth example is described in the problem statement.

Code Examples

#1 Code Example with C++ Programming

Code - C++ Programming

#include<bits/stdc++.h>
using namespace std;

int main(){
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);

  int n;
  cin>>n;
  int arr[n];
  int seats = 0;
  for(auto i = 0; i  <  n; i++){
    cin>>arr[i];
    seats += arr[i];
  }
  int party = arr[0];
  int coalition = party, k = 0, i = 0;
  int arrn[n];
  int cvheck = 0;
  while(i  <  n-1){
    i++;
    if(arr[i] <= party/2){
      coalition += arr[i];
      arrn[k] = i;
      k++;
    }
  }
  if(coalition  < = seats/2){
    cout<<0<<endl;
    return 0;
  }
  cout<<k+1<<endl;
  cout<<1<<" ";
  for(auto i = 0; i  <  k; i++>{
    cout<<arrn[i]+1
     <<" ";
  }
}

Copy The Code & Try With Live Editor

Input

–

cmd

3
100 50 50

Output

–

cmd

2
1 2

Demonstration

Codeforcess solution 1178-A A. Prime Minister C++, Java, Js and Python ,1178-A Codeforcess solution

Previous
Codeforces solution 1080-B-B. Margarite and the best present codeforces solution

Next
CodeChef solution DETSCORE - Determine the Score CodeChef solution C,C+

Try With Live Editor

Category - Codeforces Online Judge Maniruzzaman Akash 1 year ago 527 0