SRM606 Div1 Easy EllysNumberGuessing

問題概要

数当てゲームをする.数字一つを質問すると答えとその数字の絶対値が帰ってくる.質問と返答のリストが渡されるので答えを求めよ

解法

答えの候補は高々2通りなので,setに突っ込んで満たさなければeraseしていけばよい.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(ll i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(ll i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;

class EllysNumberGuessing{
public:
    int getNumber(vi a,vi b){
        int n=a.size();
        set<int> ans;
        REP(i,n){
            if(!i){
                if(0<a[i]-b[i]&&a[i]-b[i]<=1e9)ans.insert(a[i]-b[i]);
                if(0<a[i]+b[i]&&a[i]+b[i]<=1e9)ans.insert(a[i]+b[i]);
            }else{
                for(auto itr:ans){
                    if(a[i]+b[i]!=itr&&a[i]-b[i]!=itr){
                        ans.erase(itr);
                    }
                }
            }
        }
        if(ans.size()>1){
            return -1;
        }else if(!ans.size()){
            return -2;
        }else{
            return *ans.begin();
        }
    }
};