TCS Coding Questions 2022 Day 1 Slot 1

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n,j=0; cin>>n;
    int a[n]={0};
    for(int i=0;i<n;i++)
    {
        cin>>a[j];
        if(a[j]!=0) j++;
    }
    for(int i=0;i<n;i++) cout<<a[i]<<" ";
}

import java.util.*;
class Solution
{
    public static void main (String[]args)
    {
        Scanner sc = new Scanner (System.in);
        int n = sc.nextInt ();
        int arr[] = new int[n];
for(int i = 0; i < n; i++)
arr[i] = sc.nextInt ();

int count = 0;

for(int i = 0; i < n; i++)
if (arr[i] != 0)
arr[count++] = arr[i];
for(int i = count; i < n; i++)
arr[i] = 0;

for (int i = 0; i < n; i++)
System.out.print (arr[i] + " ");
}
}

n=int(input())
j=0
L=[0 for i in range(n)]
for i in range(n):
    a=int(input())
    if a!=0:
        L[j]=a
        j+=1
for i in L:
    print(i,end=" ")