forked from neetcode-gh/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
0473-matchsticks-to-square.cpp
45 lines (40 loc) · 1.23 KB
/
0473-matchsticks-to-square.cpp
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
class Solution {
int a,b,c,d;
bool fun(vector<int>& matchsticks,int i){
if(i==matchsticks.size()){
if(a==0 && b==0 && c==0 && d==0) return true;
else return false;
}
if(matchsticks[i]<=a){
a-=matchsticks[i];
if(fun(matchsticks,i+1)) return true;
a+=matchsticks[i];
}
if(matchsticks[i]<=b){
b-=matchsticks[i];
if(fun(matchsticks,i+1)) return true;
b+=matchsticks[i];
}
if(matchsticks[i]<=c){
c-=matchsticks[i];
if(fun(matchsticks,i+1)) return true;
c+=matchsticks[i];
}
if(matchsticks[i]<=d){
d-=matchsticks[i];
if(fun(matchsticks,i+1)) return true;
d+=matchsticks[i];
}
return false;
}
public:
bool makesquare(vector<int>& matchsticks) {
if(matchsticks.size()<4) return false;
int sum = accumulate(matchsticks.begin(), matchsticks.end(),0);
if(sum % 4 != 0) return false;
int sizeSum=sum/4;
a=sizeSum,b=sizeSum,c=sizeSum,d=sizeSum;
sort(matchsticks.rbegin(), matchsticks.rend());
return fun(matchsticks,0);
}
};