Cod sursa(job #196166)

Utilizator mihai_floreaFlorea Mihai Alexandru mihai_florea Data 24 iunie 2008 19:26:20
Problema Divk Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <cstring>
#include <queue>
using namespace std;
int n,k,a,b,x,s[500002],i,sol;
queue<int> r[100001];
int main(){
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    scanf("%d %d %d %d",&n,&k,&a,&b);
    for (i=1,s[0]=0;i<=n;++i){
        scanf("%d",&x);
        s[i]=(s[i-1]+x)%k;
        }
    for (i=a;i<=n;++i){
        while (!r[s[i]].empty() && r[s[i]].front()<i-b) r[s[i]].pop();
        r[s[i-a]].push(i-a);
        sol+=r[s[i]].size();
        }
    printf("%d",sol);
    return 0;
}