43 lines
1.0 KiB
C++
43 lines
1.0 KiB
C++
|
#include<stdcpp.h>
|
||
|
|
using namespace std;
|
||
|
|
class Solution{
|
||
|
|
public:
|
||
|
|
int numberOfBeams(vector<string> & bank){
|
||
|
|
int len = bank.size();
|
||
|
|
vector<string> null_bank;
|
||
|
|
for(string row_plan: bank){
|
||
|
|
bool flag = 1;
|
||
|
|
for(char ch: row_plan){
|
||
|
|
if(ch == '1'){
|
||
|
|
flag = 0;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if(!flag) null_bank.push_back(row_plan);
|
||
|
|
}
|
||
|
|
long long ans = 0;
|
||
|
|
int last = 0;
|
||
|
|
for(string row_plan: null_bank){
|
||
|
|
int num = 0;
|
||
|
|
for(char ch : row_plan){
|
||
|
|
if(ch == '1') num++;
|
||
|
|
}
|
||
|
|
ans += num*last;
|
||
|
|
last = num;
|
||
|
|
}
|
||
|
|
return ans;
|
||
|
|
}
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
int main(){
|
||
|
|
Solution sol;
|
||
|
|
vector<string> ex1 = {"011001","000000","010100","001000"};
|
||
|
|
cout<<"ex1"<<endl;
|
||
|
|
cout<<sol.numberOfBeams(ex1)<<endl;
|
||
|
|
|
||
|
|
vector<string> ex2 = {"000","000","000"};
|
||
|
|
cout<<"ex2"<<endl;
|
||
|
|
cout<<sol.numberOfBeams(ex2)<<endl;
|
||
|
|
return 0;
|
||
|
|
}
|