C. Oranges and Apples

C. Oranges and Apples
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

In 2N - 1 boxes there are apples and oranges. Your task is to choose N boxes so, that they will contain not less than half of all the apples and not less than half of all the oranges.

Input

The first input line contains one number T — amount of tests. The description of each test starts with a natural number N — amount of boxes. Each of the following 2N - 1 lines contains numbers ai and oi — amount of apples and oranges in the i-th box (0 ≤ ai, oi ≤ 109). The sum of N in all the tests in the input doesn't exceed 105. All the input numbers are integer.

Output

For each test output two lines. In the first line output YES, if it's possible to choose N boxes, or NO otherwise. If the answer is positive output in the second line N numbers — indexes of the chosen boxes. Boxes are numbered from 1 in the input order. Otherwise leave the second line empty. Separate the numbers with one space.

Sample test(s)
Input
2
2
10 15
5 7
20 18
1
0 0
Output
YES
1 3
YES
1
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
    int a,o;
    int index;
} box[2*100000+5];
int cmp(struct node aa,struct node bb)
{
    return aa.o<bb.o;
}
int main()
{
    int ccount;
    int N;
    long long oddSum;
    long long evenSum;
    long long suma;
    long long sumo;
    cin>>ccount;
    while(ccount--)
    {
        cin>>N;
        int q,w;
        suma=0;
        sumo=0;
        for(int i=0; i<2*N-1; i++)
        {
            cin>>q>>w;
            box[i].a=q;
            box[i].o=w;
            box[i].index=i+1;
            suma+=q;
            sumo+=w;
        }
        printf("YES\n");
        sort(box,box+(2*N-1),cmp);
        oddSum=0;
        evenSum=0;
        for(int i=0; i<2*N-1; i+=2)
        {
            oddSum+=box[i].a;
            if(i+1<2*N-1)
            evenSum+=box[i+1].a;
        }
        bool flag=true;
        if(oddSum>=evenSum)
        {
            for(int i=0; i<2*N-1; i+=2)
                printf("%d ",box[i].index);
            printf("\n");
            flag=false;
        }
        if(flag)
        {
            for(int i=1; i<2*N-1; i+=2)
            {
                printf("%d ",box[i].index);
            }
            printf("%d\n",box[2*N-2].index);
        }
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章