Cod sursa(job #953331)

Utilizator classiusCobuz Andrei classius Data 25 mai 2013 19:12:47
Problema Divk Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <fstream>
#include <deque>
#include <vector>

using namespace std;
ifstream f("divk.in");
ofstream g("divk.out");

int main()
{
    int n,k,a,b,s=0;
    vector<int> r;
    deque<int> dq(1,0);

    f>>n>>k>>a>>b;
    r.insert(r.begin(),k,0);
    r[0]=1;
    int sol=0;
    for(int i=1;i<=n;i++){
        int x;
        f>>x;
        s=(s+x)%k;
        //g<<s<<" "<<x<<" "<<k<<'\n';
        if(i>=a){
            if(i>=b){
                int y=dq.front();
                dq.pop_front();
                r[y]--;
            }
            sol+=r[s];
        }
        dq.push_back(s);
        r[s]++;
    }
    g<<sol;

    return 0;
}