スクリーモ総合30

#!/usr/bin/perl
use strict;

my $input = $ARGV[0];

$input =~ /^[1-9]{13}$/ or die "invalid input\n";

my @input_arr = split('', $input);

# 各牌の枚数
my @input_count;

foreach my $i (@input_arr) {
    if(!$input_count[$i]) {
	$input_count[$i] = 1;
    } else {
	$input_count[$i] += 1;
    }
}

&seek($input, $output, @input_count);

sub seek {
    (my $str, my $out, my @count) = @_;

    # 残り一枚なら単騎待ち
    if (length($str) < 2)  {
	$out .= "[$str]";
	print "$out\n";
	return;
    } 
    
    # 1~9で始まるジュンツ、コーツを探す
    for (my $i=1; $i <= 9; $i++) {
	# その数字の牌がもうなければ次
	next if $count[$i] eq 0;
	
	my ($ch1, $ch1_index, $ch2, $ch2_index, $ch3, $ch3_index) = ($i, -1, -1, -1, -1, -1);

	# テンパイ判定
	if (length($str) == 4) {
	    $ch1_index = $ch2_index = -1;
	    $ch1_index = index($str, $ch1);
	    $ch2_index = index($str, $ch1, $ch1_index + 1);
	    if ($ch1_index > -1 && $ch2_index > -1) {
		my $machi;
		for (my $j=0; $j < length($str); $j++) {
		    $machi .= substr($str, $j, 1) 
			if $j ne $ch1_index && $j ne $ch2_index;
		}

		print "$out\($ch1$ch1\)\[$machi\]\n" if &isMachiValid($machi);
	    }
	}

	#カブリ防止のためより大きい数字か同じメンツ以外は追加しない。その判定用
	$out =~ /\((\d\d\d)\)$/;
	my $heaviness = $1;

	# コーツ
	$ch2 = $ch3 = $ch1;
	$ch1_index = index($str, $ch1);
	$ch2_index = index($str, $ch2, $ch1_index + 1) if $ch1_index >= 0;
	$ch3_index = index($str, $ch3, $ch2_index + 1) if $ch2_index >= 0;
	if ($ch3_index > -1
	    && $count[$ch2] > 0 && $count[$ch3] > 0
	    && $ch1 * 100 + $ch2 * 10 + $ch3 >= $heaviness) {
	    &seekNext($str, $out, \@count, 
		     $ch1, $ch2, $ch3,
		     $ch1_index, $ch2_index, $ch3_index) ;
	}

	$ch1_index = $ch2_index = $ch3_index = -1;
	# ジュンツ
	if ($ch1 < 8 ) {
	    $ch2 = $ch1 + 1;
	    $ch3 = $ch1 + 2;
	    $ch1_index = index($str, $ch1);
	    $ch2_index = index($str, $ch1 + 1);
	    $ch3_index = index($str, $ch1 + 2);
	    
	    if ($ch2_index > -1 && $ch3_index > -1
		&& $count[$ch2] > 0 && $count[$ch3] > 0
		&& $ch1 * 100 + $ch2 * 10 + $ch3 >= $heaviness) {
		&seekNext($str, $out, \@count, 
			  $ch1, $ch2, $ch3,
			  $ch1_index, $ch2_index, $ch3_index);
	    }
	}
    }
}

sub isMachiValid {
    my $machi = @_[0];
    return 1 if (length($machi) < 2);
    return 0 if (length($machi) > 2);

    my $ch1 = substr($machi, 0, 1);
    my $ch2 = substr($machi, 1, 1);
    ($ch1, $ch2) = ($ch2, $ch1) if $ch1 > $ch2;

    return 1 if ($ch1 + 1 eq $ch2|| $ch1 + 2 eq $ch2 || $ch1 eq $ch2);
    return 0;
}

sub seekNext {
    (my $str, my $out, my $count_ref,
     my $ch1, my $ch2, my $ch3, 
     my $ch1_index, my $ch2_index, my $ch3_index) = @_;
    my @count = @$count_ref;

    my $nextstr = "";
    for (my $i=0; $i < length($str); $i++) {
	$nextstr .= substr($str, $i, 1) if $i ne $ch1_index 
	    && $i ne $ch2_index 
	    && $i ne $ch3_index;
    }

    my $nextout = "$out\($ch1$ch2$ch3\)";
    
    my @nextcount = @count;
    $nextcount[$ch1] -= 1; $nextcount[$ch2] -= 1; $nextcount[$ch3] -= 1;

    &seek($nextstr, $nextout, @nextcount);
}