AtCoder ABC 263 A問題
AtCoder ABC 263 A問題
入力に対して、ソートして、下から3つ+上から2つが等しいか、下から2つ+上から3つが等しいとすればできると思ったのですが、”1 1 2 2 2”という入力に対して、"No"と判定されてしまいます。
該当するソースコード
#include<bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
int main(){
vector<int> vec(5);
rep(i,5) cin >> vec.at(i);
sort(vec.begin(), vec.end());
//cout << vec.at(0) << vec.at(1) << vec.at(2) << vec.at(3) << vec.at(4) << endl;
if(vec.at(0) == vec.at(1) == vec.at(2) && vec.at(3) == vec.at(4)) cout << "Yes" << endl;
else if(vec.at(0) == vec.at(1) && vec.at(2) == vec.at(3) == vec.at(4)) cout << "Yes" << endl;
else cout << "No" << endl;
return 0;
}
よろしくお願いします。