Building Teams

warning: unused variable: `i`
  --> input/code.rs:36:9
   |
36 |     for i in 0..m {
   |         ^ help: if this is intentional, prefix it with an underscore: `_i`
   |
   = note: `#[warn(unused_variables)]` on by default

warning: function `take_int` is never used
 --> input/code.rs:3:4
  |
3 | fn take_int() -> i64 {
  |    ^^^^^^^^
  |
  = note: `#[warn(dead_code)]` on by default

warning: function `take_vector` is never used
 --> input/code.rs:9:4
  |
9 | fn take_vector() -> Vec<i64> {
  |    ^^^^^^^^^^^

warning: function `take_string` is never used
  --> input/code.rs:23:4
   |
23 | fn take_string() -> Vec<char> {
   |    ^^^^^^^^^^^

warning: 4 warnings emitted

use std::io::stdin;

fn take_int() -> i64 {
    let mut input = String::new();
    stdin().read_line(&mut input).unwrap();
    return input.trim().parse().unwrap()
}

fn take_vector() -> Vec<i64> {
    let mut input = String::new();
    stdin().read_line(&mut input).unwrap();
    let arr = input.trim().split_whitespace().map(|x| x.parse().unwrap()).collect();
    return arr;
}

fn take_vector_usize() -> Vec<usize> {
    let mut input = String::new();
    stdin().read_line(&mut input).unwrap();
    let arr = input.trim().split_whitespace().map(|x| x.parse().unwrap()).collect();
    return arr;
}

fn take_string() -> Vec<char> {
    let mut input = String::new();
    stdin().read_line(&mut input).unwrap();
    let vec:Vec<char> = input.trim().chars().collect();
    return vec;
}

pub fn main() {
    let line = take_vector_usize();
    let (n, m) = (line[0], line[1]);
    let mut team = vec![0; n];
    let mut e: Vec<Vec<usize>> = vec![vec![]; n];

    for i in 0..m {
        let line = take_vector_usize();
        let (mut a, mut b) = (line[0], line[1]);
        (a, b) = (a - 1, b - 1);
        e[a].push(b);
        e[b].push(a);
    }

    let mut visited = vec![false; n];
    let mut visited_count = 0;
    let mut nextt = 0;
    while visited_count < n {
        let mut work = vec![nextt];
        team[nextt] = 0;
        while work.len() > 0 {
            let cur = work.pop().unwrap();
            visited[cur] = true;
            visited_count += 1;
            let cur_team = team[cur];
            for friend in &e[cur] {
                if !visited[*friend] {
                    team[*friend] = 1 - cur_team;
                    work.push(*friend);
                } else {
                    if team[*friend] == cur_team {
                        println!("IMPOSSIBLE");
                        return;
                    }
                }
            }
        }
        while nextt < n && visited[nextt] {
            nextt += 1
        }
    }
    for t in team {
        print!("{} ", t + 1);
    }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED