HDU 5776 sum 【思维 + 前缀和】

传送门
题意: 给定一个长度为n的序列, 问是否存在一个连续子序列使得他们的和是给定的m的倍数.

思路: 很明显, 一提到连续区间的和, 那肯定是和前缀和有关的, 即我们需要找是否至少存在两个在%m意义下相等的数字即可. 为什么%m的不用讲吧? 就是因为超过了m后算倍数是等价于%m的, 所以我们直接做一遍前缀和, 边做边%, 然后标记一些之前出现过的前缀和, 一旦在此出现那么这两个之间的序列相加就是m的倍数.

AC Code

const int maxn = 1e5+5;
bool vis[maxn];
int sum[maxn];
void solve()
{
    int n,m;
    scanf("%d%d",&n, &m);
    Fill(vis, 0); Fill(sum, 0);
    int flag = 0; vis[0] = 1;
    for(int i = 1 ; i <= n ; i ++){
        int u; scanf("%d",&u);
        sum[i] = (sum[i-1] + u ) % m;
        if(vis[sum[i]]) flag = 1;
        else vis[sum[i]] = 1;
    }
    printf("%s\n", flag ? "YES": "NO");
}

那么万一 u 可以是负数了, 也就是倍数可以-x, 其实也很简单, 我们转化为正数就行, 只不过此时不能用sum转了, 因为此时为 + k ) % k, 那么用sum就会改变这个前缀值造成错误, 所以我们需要另开变量来转即可. 如 LeetCode 974
AC Code

class Solution {
public:
    int subarraysDivByK(vector<int>& A, int K) {
        int sum = 0, ans = 0;
        map<int, int>mp; mp[0] = 1;
        for (int i = 0 ; i < A.size() ; ++ i) {
            sum += A[i]; sum %= K;
            int tmp = (sum % K + K) %K; // 另开变量%
            ans += mp[tmp];
            mp[tmp]++;
        }
        return ans;
    }
};