Cod sursa(job #1335021)

Utilizator PikachuPikachu Pikachu Data 4 februarie 2015 21:06:46
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <cstdio>
 
using namespace std;
 
int s[500005], f[500005];
 
int main() {
    freopen("divk.in", "r", stdin);
    freopen("divk.out", "w", stdout);
 
    int n, k, a, b, x, aux;
 	long long sol = 0;

    scanf("%d%d%d%d", &n, &k, &a, &b);
 
    for(int i = 1; i <= n; ++ i) {
        scanf("%d", &x);
        s[i] = (s[i - 1] + x) % k;
    }

    ++ f[0];

    for(int i = a; i <= n; ++ i) {
    	sol += f[s[i]];
    	++ f[s[i - a + 1]];
    	if(i - b >= 0)
    		-- f[s[i - b]];
    }

    printf("%lld\n", sol);
 
    return 0;
}