zoj 1909 Square

1. #include<iostream>
2. #include<algorithm>
3. using namespace std;
5. int  len[20];
6. int  average;
7. int  n;
8. int  occ[20];
10. bool formSquare(int i, int j, int  amount)
11. {
12.     if (amount < 0 || j < 0) return false;
13.     if (amount == 0) {
14.         if (i==2) return true;
15.         return formSquare(i+1, n-1, average);
16.     }
17.     
18.     if (!occ[j])
19.     {
20.         occ[j] = 1;
21.         if (formSquare(i, j-1, amount - len[j]) ) return true;
22.         occ[j] = 0;
23.         while (j && len[j] == len[j-1]) j--;
24.     }
25.     return formSquare(i, j-1, amount);
26. }
28. int main()
29. {
30.     int k;
31.     cin>>k;
32.     for (int i=0; i<k; i++)
33.     {
34.         cin>>n;
35.         average = 0;
36.         for (int j=0; j<n; j++)
37.         {
38.             cin>>len[j];
39.             average += len[j];
40.         }
41.         if (average%4) {
42.             cout<<"no"<<endl;
43.             continue;
44.         }
45.         
46.         sort(len, len+n);
47.         average >>= 2;
48.         if (len[n-1] > average)
49.         {
50.             cout<<"no"<<endl;
51.             continue;
52.         }
53.         for (int j=0; j<n; j++) occ[j] = 0;
54.         bool done = formSquare(0, n-1, average);
56.         if (done) cout<<"yes"<<endl;
57.         else cout<<"no"<<endl;
58.     }
59. }