Concours de programmation AtCoder dwango B à résoudre en Ruby, Perl et Java

introduction

• Utilisez les recommandations dans AtCoder Problems * pour résoudre les problèmes passés. Merci à AtCoder et AtCoder Problems.

Ce thème

• Concours de programmation AtCoder dwango B - Chaîne de caractères Niko Niko * Difficulty: 832

Le thème de cette époque est l'agrégation par réduction Ruby Par exemple, la chaîne de caractères «1251251252525» est convertie en «25» -> «a» pour faciliter le comptage des caractères. => 1a1a1aaa Le nombre de «a» consécutifs est (1, 1, 3), donc additionnez le nombre de combinaisons pour chacun.

bit.rb

s = gets.chomp

s.gsub!(/25/, 'a')
p = []
c = 0
0.upto(s.size - 1) do |i|
  if s[i] == 'a'
    c += 1
  elsif c > 0
    p.push(c)
    c = 0
  end
  p.push(c) if i == s.size - 1 && c > 0
end
sum = 0
0.upto(p.size - 1) do |i|
  sum += (1..p[i]).reduce{|a, b| a + b}
end
puts sum

reduce.rb

  sum += (1..p[i]).reduce{|a, b| a + b}
  sum += (1..p[i]).reduce(&:+)
  sum += (1..p[i]).inject(:+)
  sum += (1..p[i]).sum

La partie de réduction peut être soit injectée, soit additionnée (ruby 2.4 ou version ultérieure). ** Addenda ** Ajout de réduire (&: +) dans les commentaires.

Perl

perl.pl

use v5.18; # strict say state
use List::Util qw/reduce/;

chomp (my $s = <STDIN>);
$s =~ s/25/a/g;
my $c = 0;
my @p;
for my $i (0..length($s)-1) {
  if (substr($s, $i, 1) eq 'a') {
    $c++;
  } elsif ($c > 0) {
    push @p, $c;
    $c = 0;
  }
  if ($i == length($s)-1 && $c > 0) {
    push @p, $c;
  }
}
my $sum = 0;
for my $i (0..$#p) {
  $sum += reduce {$a + $b} 1..$p[$i];
}
say $sum;

reduce.pl

use List::Util qw/reduce sum/;

  $sum += reduce {$a + $b} 1..$p[$i];
  $sum += sum(1..$p[$i]);

La partie réduite peut également être une somme. Java

java.java

import java.util.*;
import java.util.stream.IntStream;

class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String s = sc.next();
        sc.close();
        s = s.replace("25", "a");
        List<Integer> p = new ArrayList<>();
        int c = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.substring(i, i + 1).equals("a")) {
                c++;
            } else if (c > 0) {
                p.add(c);
                c = 0;
            }
            if (i == s.length() - 1 && c > 0) {
                p.add(c);
            }
        }
        int sum = 0;
        for (int i = 0; i < p.size(); i++) {
            sum += IntStream.rangeClosed(1, p.get(i)).reduce(0, (a, b) -> a + b);
        }
        System.out.println(sum);
    }
}

reduce.java

            sum += IntStream.rangeClosed(1, p.get(i)).reduce(0, (a, b) -> a + b);
            sum += IntStream.rangeClosed(1, p.get(i)).sum();

La partie réduite peut également être une somme.

Résumé

• Concours de programmation Dwango résolu B
• Familiarisez-vous avec Ruby
• Familiarisez-vous avec Perl
• Familiarisez-vous avec Java
• Se familiarise avec réduire

Site référencé

• La meilleure façon de additionner les éléments d'un tableau dans Ruby *