CodeIQ：プラマイゼロは何通り？

問題の概要

次のようにパラメータが入力されます。

入力された値から2つを選んだとき、それらの和がゼロになるような組み合わせが何通りあるかを求めてください。

私のプログラム

C++で解答しています。

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <set>
using namespace std;

int check(set< int >& to_add, set< int >& to_search, int n){
	int	ret = 0;

	set< int >::iterator i = to_search.find(-1*n);
	if(i != to_search.end()){
		ret = 1;
	}

	to_add.insert(n);

	return ret;
}

int isNumber(char* str){
	if((str[0] == '\n') || (str[0] == '\r') || (str[0] == 0)){
		return 0;
	}

	for(int i=0; str[i]; i++){
		if((str[i] == '\n') || (str[i] == '\r') || (str[i] == 0)){
			str[i] = 0;
			break;
		}
		else if((str[i] == '+') || (str[i] == '-')){
			continue;
		}
		else if(isdigit(str[i]) == 0){
			return 0;
		}
	}
	return 1;
}

int main(int argc, char* argv[]){
	char str[1024] = {0};
	set< int >	MinusSet;
	set< int >	PlusSet;
	int	Count = 0;

	while( fgets(str, sizeof(str), stdin) != NULL ){
		if(isNumber(str) == 0){
			break;
		}

		int n = atoi(str);
		if(n<0){
			Count += check(MinusSet, PlusSet, n);
		}
		else{
			Count += check(PlusSet, MinusSet, n);
		}
	}
	printf("%d", Count);
	return 0;
}

解説

それでは解説します。
言語にC++を選択したのはsetが使いたかったからです。この問題を解答した時点ではPythonを覚えていないので選択肢としてはC++かJava（JavascriptやPHPでも連想配列で代用できますがうっとおしいので）ですが、わざわざクラスを作るのが面倒なのでC++にしました。なのでプログラムはC++と言うよりsetを使ってC言語で書いたという感じになっています。

雑感

それほど変わったテクニックなどは使っていません。ただ、問題の通り合計して0に成る値を探すよりも、setのようなデータ構造を利用して符合を反転して同じ値を探す方が簡単なプログラムになります。私の経験上setは意外に使い出のあるデータ構造です。