分析外星訊號

您的好友急速上線發表於2018-01-20

題意:給出長度為N的訊號紀錄時,編寫程式計算總和等於k的子序列個數

通過公式來生成輸入值

通過線上演算法來程式設計,程式碼如下:

#include <bits/stdc++.h>

using namespace std;

struct RNG
{
    unsigned seed;
    RNG():seed(1983) {}
    unsigned next()
    {
        unsigned ret=seed;
        seed=((seed*214013u)+2531011u);
        return ret%10000+1;
    }
};

int countRartialSums(int k,int n)
{
    RNG rng;
    int ret=0;
    long long pusm=0;
    queue<long long>pusms;
    for(int i=0; i<n; i++)
    {
        pusm+=rng.next();
        pusms.push(pusm);
        while(pusms.front()+k<pusm)
            pusms.pop();
        if(pusms.front()+k==pusm)
            ++ret;
    }
    return ret;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int k,n;
        scanf("%d %d",&k,&n);
        int sum=countRartialSums(k,n);
        printf("%d\n",sum);
    }
    return 0;
}



相關文章