b5i1mX - Online C++ Compiler & Debugging Tool

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int main() {
    int N;
    scanf("%d", &N);

    for (int i = 0; i < N; i++) {
        int m,num;
        int sum = 0;
		scanf("%d", &m);
		scanf("%d", &num);
		int *A = (int *)malloc(m * sizeof(int));
		for(int i = 0 ;i < m;i++)
			scanf("%d",&A[i]);
		for(int j = 0 ; j < m;j++){ //奇数
			if(A[j] == num){
				int wide = 0;
				while(j - wide >=0 && j+wide <= m-1){
					sum++;
					wide ++;
				}
			}

		}
		for(int j = 0 ; j < m - 1 ;j++){ //偶数
			if(A[j] == num && A[j+1] == num){
				int wide = 0;
				while(j - wide >=0 && j+1+wide <= m-1){
					sum++;
					wide ++;
				}
			}

		}
		printf("%d\n",sum);
    }

    return 0;
}