My issue
output is coming wrong
My code
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main() {
int t,v;
cin>>t;
while(t--){
int n=0;cin>>n;
vector<int>v(n);
for(int i=0;i<n;++i)
{
cin>>v[i];
}
vector<int>prefix(n);
prefix[0]=v[0];
for(int i=1;i<n;++i)
{
prefix[i]=(prefix[i-1],v[i]);
}
vector<int>suffix(n);
suffix[n-1]=v[n-1];
for(int i=n-2;i>=0;--i)
{
suffix[i]=(suffix[i+1],v[i]);
}
int strong =0;
for(int i=0;i<n;++i)
{
if(i==0 && suffix[i]>i){
strong++;
}
else if((prefix[i-1],suffix[i+1])>1)
{
strong++;
}
}
cout<<strong<<'\n';
}
}
Learning course: Number theory
Problem Link: CodeChef: Practical coding for everyone