Основное
 
 
Отправитель Задача Язык Вердикт Время Память Отослано Протест.  
222190799 Дорешивание:
urirob
1609C - 48 C++20 (GCC 11-64) Полное решение 764 мс 756 КБ 2023-09-07 13:49:18 2023-09-07 13:49:18
→ Исходный код
#include <bits/stdc++.h>
#define ll long long
using namespace std;
 
bool prost(int n){
	if(n==1) return false;
	for(int i=2;i*i<=n;i++)
		if(n%i==0) return false;
	return true;
}
int main(){
	int t; cin>>t;
	while(t--){
	int n,k; cin>>n>>k;
	int a[n];
	long long s=0;
	for(int i=0;i<n;i++) cin>>a[i];
	for(int i=0;i<n;i++){
		if(prost(a[i])){
			long long r=1,l=1;
			while(r*k+i<n&&a[r*k+i]==1) r++;
			while(i-l*k>=0&&a[i-l*k]==1) l++;
			l--;r--;
			s+=l+r+l*r;
		}
	}
	cout<<s<<endl;
	}
	return 0;
}
?
Время: ? ms, память: ? КБ
Вердикт: ?
Ввод
?
Вывод участника
?
Ответ жюри
?
Комментарий чекера
?
Диагностика
?
Показать детали тестирования