c_cpp HR冰淇淋店

Posted 2021-05-14

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

// #Maps #Hashing #HR #InterviewPreparationKit
// https://www.hackerrank.com/challenges/ctci-ice-cream-parlor/problem?h_l=interview&playlist_slugs%5B%5D=interview-preparation-kit&playlist_slugs%5B%5D=search

void whatFlavors(vector<int> a,int money){
    int n=a.size();
    map < int, vector<int> > m;
    for(int i=0;i<n;i++){
        m[a[i]].push_back(i+1);
    }
    for(int i=0;i<n;i++){
        int x=money-a[i];
        int equal=0;
        if(x==a[i]){
            equal=1;
        }
            if( (equal==0 && m[x].size()!=0) || (equal==1 && m[x].size()!=1) ){
               if(m[x].size()==1){
                   if(i+1>m[x][0]){
                       cout<<m[x][0]<<" "<<i+1<<endl;
                   }else{
                       cout<<i+1<<" "<<m[x][0]<<endl;
                   }
               }else{
                   cout<<i+1<<" "<<m[x][1]<<endl;
               }
               break;

            }
        
    }
}

int main()
{
   int t;
   cin>>t;
   while(t--){
       int money;
       cin>>money;
       int n;
       cin>>n;
       vector<int> cost(n);
       for(int i=0;i<n;i++){
           cin>>cost[i];
       }
       whatFlavors(cost,money);
   }

    return 0;
}