ABC147-F-数论,思维

题目大意:

给你一个等差数列$a_i=x+i\ast d$。然后你可以从里面选出一个子集$S$(包含0和全集)。问你能够得到多少不同的结果.

$n\le 1e5,x,d\le 1e8$

题目思路：

首先，特判$d=0$.若$x=0$.则答案为0.否则答案为$n+1$.

1.考虑选择了$k$个元素出来。那么$sum=k\ast x+c\ast d$. $kx$定值，那么c的范围一定是: $0+...+k-1\le c\le n-k+...+n-1$并且连续.

2.映射到坐标轴上，就是位于直线$y=kx+b$的离散点。我们的目的就是要统计这些离散点的个数.这里用到了一个技巧:

由于点之间严格相差$d$.所以我们将其$/d$之后，他们的增量变为1.那这就变成了一条线段。求出左右两个端点即可求出点的个数。

3.1 而且可以发现,对于一些$k$来讲,他们的点会发生重合。例如:

$x=3,d=2$. 那么
$k=1时$点为:3 5 7 9 11 13 15…
$k=3时$点为:15 17…

两个在15处重合，往后的都一样了.

3.2 并且对于一些$k$来讲,他们的点永远不会发生重合。起点不重合，且差值相同。那么永远不会相遇。

具体的，当$k_i\ast x\equiv k_j\ast x(modd)$时，他们有点会重合.那么对于在一个同余类的$k$.我们统一处理,$/d$之后变线段，丢到set里求线段并即可.

4.但是d太大了，这里需要进一步优化。

有两种解决方法:

方法一:

直观的，枚举$k_i\in [0,d)$,然后再枚举$k_i+jd\le n$.这些数一定在一个同余类中。但是这样并不一定全面.因为有可能存在$k_d\ne k_i+jd$的$k_d$使得他们两个也在同一个同余类中。本质就是因为$x,d$不互质.
但是我们只要将$x,d$除以$gcd$.将其转化为互质的。这样以来。就是以$k$为同余类了.然后就很显然了.具体细节见代码.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int main()
{
    ios::sync_with_stdio(false);
    ll n , x , d; cin >> n >> x >> d;
    if (d == 0){
        if (x == 0) cout << 1 << endl;
        else {
            cout << n + 1 << endl;
        }
        return 0;
    }
    ll gcd = __gcd(x , d);
    x /= gcd; d /= gcd;
    if (d < 0) {
        x = -x; d = -d;
    }
    ll ans = 0;
    for (ll i = 0 ; i < d ; i++){
        set<pair<ll,ll>>s;
        for (ll j = i ; j <= n ; j += d){
            ll tmp = x * j;
            ll l = (0+j-1)*j/2 , r = (n-1+n-j)*j/2;
            s.insert(mp(tmp/d + l , tmp/d + r));
        }
        ll mi , mx;
        mi = mx = -1e17;
        mx--;
        for (auto g : s){
            if (g.first > mx){
                ans += mx - mi + 1;
                mi = g.first;
                mx = g.second;
            }else {
                mx = max (mx , g.second);
            }
        }
        ans += mx - mi + 1;
    }
    cout << ans << endl;
    return 0;
}

方法二:(未验证)

直接枚举$k\in [0,n]$.利用$modd$归桶。然后再做$set$的线段合并。可以map套set或者离散化余数再set吧？

最多是开$1e5$个set,每个set里一个线段?但是随机数据下应该不会跑满，说不定还很小.