Saturday 16 November 2019

B. Rooms and Staircases/Codeforces Round #592 (Div. 2)

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 n 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 1 and n).
If Nikolay is currently in some room, he can move to any of the neighbouring rooms (if they exist). Rooms with numbers i and i+1 on each floor are neighbouring, for all 1in1. There may also be staircases that connect two rooms from different floors having the same numbers. If there is a staircase connecting the room x on the first floor and the room x on the second floor, then Nikolay can use it to move from one room to another.
The picture illustrates a house with n=4. There is a staircase between the room 2 on the first floor and the room 2 on the second floor, and another staircase between the room 4 on the first floor and the room 4 on the second floor. The arrows denote possible directions in which Nikolay can move. The picture corresponds to the string "0101" in the input.
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 t (1t100) — 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 n (1n1000) — the number of rooms on each floor.
The second line contains one string consisting of n characters, each character is either a '0' or a '1'. If the i-th character is a '1', then there is a staircase between the room i on the first floor and the room i on the second floor. If the i-th character is a '0', then there is no staircase between the room i on the first floor and the room i on the second floor.
In hacks it is allowed to use only one test case in the input, so t=1 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 6 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;
}