B. Rooms and Staircases
time limit per test
1 second
memory limit per test
256 megabytes
input
standard input
output
standard output
Nikolay lives in a two-storied house. There are rooms on each floor, arranged in a row and numbered from one from left to right. So each room can be represented by the number of the floor and the number of the room on this floor (room number is an integer between and ).
If Nikolay is currently in some room, he can move to any of the neighbouring rooms (if they exist). Rooms with numbers and on each floor are neighbouring, for all . There may also be staircases that connect two rooms from different floors having the same numbers. If there is a staircase connecting the room on the first floor and the room on the second floor, then Nikolay can use it to move from one room to another.
Nikolay wants to move through some rooms in his house. To do this, he firstly chooses any room where he starts. Then Nikolay moves between rooms according to the aforementioned rules. Nikolay never visits the same room twice (he won't enter a room where he has already been).
Calculate the maximum number of rooms Nikolay can visit during his tour, if:
- he can start in any room on any floor of his choice,
- and he won't visit the same room twice.
Input
The first line of the input contains one integer () — the number of test cases in the input. Then test cases follow. Each test case consists of two lines.
The first line contains one integer — the number of rooms on each floor.
The second line contains one string consisting of characters, each character is either a '0' or a '1'. If the -th character is a '1', then there is a staircase between the room on the first floor and the room on the second floor. If the -th character is a '0', then there is no staircase between the room on the first floor and the room on the second floor.
In hacks it is allowed to use only one test case in the input, so should be satisfied.
Output
For each test case print one integer — the maximum number of rooms Nikolay can visit during his tour, if he can start in any room on any floor, and he won't visit the same room twice.
Example
input
Copy
4 5 00100 8 00000000 5 11111 3 110
output
Copy
6 8 10 6
Note
In the first test case Nikolay may start in the first room of the first floor. Then he moves to the second room on the first floor, and then — to the third room on the first floor. Then he uses a staircase to get to the third room on the second floor. Then he goes to the fourth room on the second floor, and then — to the fifth room on the second floor. So, Nikolay visits rooms.
There are no staircases in the second test case, so Nikolay can only visit all rooms on the same floor (if he starts in the leftmost or in the rightmost room).
In the third test case it is possible to visit all rooms: first floor, first room second floor, first room second floor, second room first floor, second room first floor, third room second floor, third room second floor, fourth room first floor, fourth room first floor, fifth room second floor, fifth room.
In the fourth test case it is also possible to visit all rooms: second floor, third room second floor, second room second floor, first room first floor, first room first floor, second room first floor, third room.
Code:
#include<bits/stdc++.h>
using namespace std;
int main()
{
char ar[1009];
long long i,j,n,t,a,mx;
while(cin>>t)
{
for(i=1;i<=t;i++)
{
cin>>n;
cin>>ar+1;
mx=-1;
for(j=1;j<=n;j++)
{
if(ar[j]=='1')
{
a=max(j-1,n-j)+1;
//cout<<"a="<<a<<endl;
if(mx<a)
mx=a;
}
}
if(mx>=0)
cout<<mx*2<<endl;
else
cout<<n<<endl;
}
}
return 0;
}