Come eseguire iterazioni sul prodotto di più intervalli o iteratori?

Question

Esiste un modo naturale in Rust per ripetere il "prodotto" di più intervalli o iteratori?

Viene fuori quando si sta ripetendo un array multidimensionale o forse uno spazio di stato. Ad esempio, voglio considerare tutti i valori possibili di una tupla booleana con 5 elementi. L'annidamento dei loop 5 for è un po 'ingombrante.

Answer 1

Ecco una macro che fa il lavoro:

#![feature(macro_rules)] 

macro_rules! product(
    ($first:ident, $($next:ident),*) => (
     $first.iter() $(
      .flat_map(|e| std::iter::Repeat::new(e) 
       .zip($next.iter())) 
     )* 
    ); 
) 

fn main() { 
    let a = ['A', 'B', 'C']; 
    let b = [1i, 4]; 
    let c = [true, false]; 
    let d = ['x', 'y']; 

    for (((a, b), c), d) in product![a, b, c, d] { 
     println!("{} {} {} {}", a, b, c, d); 
    } 
} 

uscita:

A 1 true x 
A 1 true y 
A 1 false x 
A 1 false y 
A 4 true x 
A 4 true y 
etc... 

Playpen example

La macro restituisce il seguente

a.iter() 
    .flat_map(|e| std::iter::Repeat::new(e).zip(b.iter())) 
    .flat_map(|e| std::iter::Repeat::new(e).zip(c.iter())) 
    .flat_map(|e| std::iter::Repeat::new(e).zip(d.iter())) 

flat_map(|e| ...) combina una sequenza di iteratori in un iteratore. Il e è un elemento prodotto da un iteratore.

std::iter::Repeat::new(e) crea un iteratore che si ripete e.

.zip(...) itera su due iteratori contemporaneamente, producendo gli elementi di entrambi in coppia.

macro sono un po 'più lungo da spiegare, quindi è meglio leggere il macro guide

non so se questo è il modo migliore per farlo.

Answer 2

Il itertools crate ha una macro molto ergonomica (iproduct!) per iterare sul prodotto degli iteratori. Ecco un esempio:

#[macro_use] 
extern crate itertools; 

pub fn main() { 
    let a = ['A', 'B', 'C']; 
    let b = [1, 4]; 
    let c = [true, false]; 
    let d = ['x', 'y']; 

    for (a, b, c, d) in iproduct!(&a, &b, &c, &d) { 
     println!("{} {} {} {}", a, b, c, d); 
    } 
}