AtCoder Beginner Contest 146_E - Rem of Sum is Num

预处理即可
我们要找的是 (f[i] - f[j]) % k == i - j
移项可得 f[i] - i = f[j] - ji - j <= k 的条件下
因此题目变成了,对于每个右端点,在它的左边 k - 1 个有多少个满足 f[i] - i = f[j] - j
f[i] 是前缀和数组
AC_CODE

#include <map>
#include <iostream>
#define LL long long

using namespace std;

const int N = 2e5 + 10;

LL f[N];
map<LL, int> mp;
LL ans;
int main() {
    
    int n, k; cin >> n >> k;
    for(int i = 1; i <= n; i ++ ) {
        cin >> f[i]; f[i] += f[i - 1];
    }
    for(int i = 1; i <= n; i ++ )   
        f[i] = (f[i] - i) % k;
    int idx = min(n, k - 1);
    for(int i = 0; i <= idx; i ++ ) {
        ans += mp[f[i]];
        mp[f[i]] ++;
        // cout << ans << endl;
    }
    for(int i = idx + 1; i <= n; i ++ ) {
        mp[f[i - k]] --;
        ans += mp[f[i]];
        mp[f[i]] ++;
    }
    cout << ans << endl;
    return 0;
}
上一篇:leetcode 经典题146 LRU cache Doublelinked list+Hashmap解法


下一篇:study17