Cod sursa(job #196153)

Utilizator mihai_floreaFlorea Mihai Alexandru mihai_florea Data 24 iunie 2008 18:37:01
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.52 kb
#include <cstring>
#include <queue>
using namespace std;
int n,k,a,b,x,s[500001],i,j,sol;
queue<int> r[100000];
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]].front()<i-b) r[s[i]].pop();
        r[s[i+1-a]].push(i+1-a);
        sol+=r[s[i]].size();
        }
    printf("%d",sol);
    return 0;
}