ZJUT12 Day4

ZJUT12 Day4

1、CF 1389E Calendar Ambiguity

tags：数学

这题要掌握同余一个非常关键的性质：$ac\equiv bc(modp)$可以转化为$a\equiv b(mod\frac{p}{gcd(p,c)})$。

那么分析题目，容易得到如下同余式：

$(y-1)d+x\equiv (x-1)d+y(modw),1\le x<y\le min(m,d)$

移项，

$(y-x)(d-1)\equiv 0(modw)$

这时套用最上面给出的公式得到$y-x\equiv 0(mod\frac{w}{gcd(w,d-1)})$

下面记$p=\frac{w}{gcd(w,d-1)},up=min(m,d)$。

那么我们就是要找到在$[1,up]$区间内所有$y-x$是$p$的倍数的整数对$(x,y)(x<y)$。

最简单的想法就是枚举$y-x$，从$p$枚举到$cnt=\lfloor \frac{up}{p}\rfloor$。

那么答案就是${\Sigma }_{i=1}^{cnt}(up-pi)$，化简一下之后变成$cnt\ast up-p(1+cnt)cnt/2$，$O(1)$即可求得答案。

算上之前的$gcd$，复杂度为$O(log(n))$。

做的时候推到最后一步不会求对数了。实际上没理解到$x,y$的取值范围是固定的，知道取值范围不变之后就非常好做了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
    int t; cin >> t;
    while (t--)
    {
        ll m, d, w; cin >> m >> d >> w;
        ll p = w / __gcd(w, d - 1);
        ll up = min(m, d), cnt = up / p;
        cout << up * cnt - p * (1 + cnt) * cnt / 2 << endl;
    }
    return 0;
}

2、CF 474D MUH and Cube Walls

tags：字符串，KMP

实际上就是在给的$n$个数里面找有没有出现和下面的$w$个数一样的形状。

那么做个差分跑一遍KMP就完了。

注意特判$w=1$。

复杂度$O(n+m)$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main()
{
    int n, w; cin >> n >> w;
    if (w == 1) cout << n << endl;
    else
    {
        vector <int> a(n), b(w);
        vector <int> da(n), db(w);
        for (int i = 0; i < n; ++i) cin >> a[i];
        for (int i = 0; i < w; ++i) cin >> b[i];
        for (int i = 1; i < n; ++i) da[i] = a[i] - a[i - 1];
        for (int i = 1; i < w; ++i) db[i] = b[i] - b[i - 1];
        vector <int> nxt(w, 0);
        for (int i = 2, j = 0; i < w; ++i)
        {
            while (j > 0 && db[j + 1] != db[i]) j = nxt[j];
            if (db[j + 1] == db[i]) ++j;
            nxt[i] = j;
        }
        int cnt = 0;
        for (int i = 1, j = 0; i < n; ++i)
        {
            while (j > 0 && db[j + 1] != da[i]) j = nxt[j];
            if (db[j + 1] == da[i]) ++j;
            if (j == w - 1) ++cnt, j = nxt[j];
        }
        cout << cnt << endl;
    }
    return 0;
}