小花梨的三角形

这道题目其实不难
从上往下搜索一边，
从下往上搜索一边，
也就是要让把所有的等边三角形都找出来，然后再排序，放入set中，输出个数。
但从下往上有一点我一直没注意，有一种情况没有搜索，
还是要观察仔细啊

#include <stdio.h>
#include <string>
#include <algorithm>
#include <map>
#include <set>
#include <iostream>
using namespace std;
string str[105];
set<string>w;
int n;
void fun(char a,char b,char c)
{
	char s[5];
	s[0] = 3;
	s[0] = a,s[1] = b,s[2] = c;
	sort(s,s+3);
	w.insert(s);
	return;
}
int main()
{
	scanf("%d",&n);
	getchar();
	for(int i = 0;i < n+1;i++){
		cin >> str[i];
	}
	
	for(int i = 0;i < n;i++){
		for(int j = 0;j < str[i].size();j++){
			for(int k = 1;i+k < n+1;k++){
				fun(str[i][j],str[i+k][j],str[i+k][j+k]);
			//	printf("%c %c %c\n",str[i][j],str[i+k][j],str[i+k][j+k]);
			}
		}
	}
	
	for(int i = n;i >= 2;i--){
		for(int j = 1;j < str[i].size()-1;j++){
			for(int k = 1;j-k>=0 && str[i-k][j] != '\0';k++){
				fun(str[i][j],str[i-k][j-k],str[i-k][j]);
			//	printf("%c %c %c\n",str[i][j],str[i-k][j-k],str[i-k][j]);
			}
		}
	}
	printf("%d",w.size());
	return 0;
}