-
-
Notifications
You must be signed in to change notification settings - Fork 8.8k
/
Copy pathSolution.rs
53 lines (49 loc) · 1.53 KB
/
Solution.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
struct Bank {
balance: Vec<i64>,
}
/**
* `&self` means the method takes an immutable reference.
* If you need a mutable reference, change it to `&mut self` instead.
*/
impl Bank {
fn new(balance: Vec<i64>) -> Self {
Bank { balance }
}
fn transfer(&mut self, account1: i32, account2: i32, money: i64) -> bool {
let (account1, account2, n) = (account1 as usize, account2 as usize, self.balance.len());
if n < account1 || n < account2 {
return false;
}
if self.balance[account1 - 1] < money {
return false;
}
self.balance[account1 - 1] -= money;
self.balance[account2 - 1] += money;
true
}
fn deposit(&mut self, account: i32, money: i64) -> bool {
let (account, n) = (account as usize, self.balance.len());
if n < account {
return false;
}
self.balance[account - 1] += money;
true
}
fn withdraw(&mut self, account: i32, money: i64) -> bool {
let (account, n) = (account as usize, self.balance.len());
if n < account {
return false;
}
if self.balance[account - 1] < money {
return false;
}
self.balance[account - 1] -= money;
true
}
}/**
* Your Bank object will be instantiated and called as such:
* let obj = Bank::new(balance);
* let ret_1: bool = obj.transfer(account1, account2, money);
* let ret_2: bool = obj.deposit(account, money);
* let ret_3: bool = obj.withdraw(account, money);
*/