牛客练习赛44 B：小y的线段

链接：https://ac.nowcoder.com/acm/contest/634/B 来源：牛客网

题目描述

给出$n$条线段，第$i$条线段的长度为$a_i$，每次可以从第$i$条线段的j位置跳到第$i + 1$条线段的$j+1$位置。如果第$i+1$条线段长度不到$j+1$，那么就会回到第$i$条线段的$0$位置，然后继续跳。 问从第$i$条线段的$0$位置跳到第$n$条线段需要跳多少次 为了减少输入量，a数组将由以下方式得到

unsigned int SA, SB, SC;
 int mod;
 unsigned int Rand(){
 SA ^= SA << 16;
 SA ^= SA >> 5;
 SA ^= SA << 1;
 unsigned int t = SA;
 SA = SB;
 SB = SC;
 SC ^= t ^ SA;
 return SC;
 }
 int main() {
 cin>>n>>mod>>SA>>SB>>SC;
 for(int i = 1;i <= n;++i) a[i] = Rand() % mod + 1;
 }

输入描述:

第一行两个正整数$n,mod$，表示一共有$n$条线段

第二行$3$个数字，分别为$SA,SB,SC$

输出描述:

一行一个数字，表示从每条线段跳到$n$的次数之和。 示例1 输入

5 5 5 6 4

输出

13

备注:

$1leq n leq 2 imes 10^7$

$1 leq mod leq 6662333$

思路

首先我们可以计算出如果没有回到第$i$条线段$0$位置这个操作的次数和：(ans=n imes (n-1)/2)

然后我们只需找出需要跳回$0$位置的点的位置$i$，在位置$i$之前的点都需要多一次跳的操作，所以需要加上$i-1$次

代码

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define ms(a,b) memset(a,b,sizeof(a))
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const int maxn=3e7+10;
// const int mod=1e9+7;
const int maxm=1e3+10;
using namespace std;
unsigned int SA, SB, SC;
int mod,n;
int a[maxn];
unsigned int Rand()
{
    SA ^= SA << 16;
    SA ^= SA >> 5;
    SA ^= SA << 1;
    unsigned int t = SA;
    SA = SB;
    SB = SC;
    SC ^= t ^ SA;
    return SC;
}
int main(int argc, char const *argv[])
{
    #ifndef ONLINE_JUDGE
        freopen("/home/wzy/in.txt", "r", stdin);
        freopen("/home/wzy/out.txt", "w", stdout);
        srand((unsigned int)time(NULL));
    #endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>mod>>SA>>SB>>SC;
    for(int i = 1;i <= n;++i) a[i] = Rand() % mod + 1;
    // 如果不回到0，有多少
    ll ans=1LL*(n-1)*n/2;
    int res=a[n];
    // 从后往前查找，
    for(int i=n-1;i>=1;i--)
    {
        res--;
        res=min(res,a[i]);
        if(!res)
        {
            ans+=(i-1);
            res=a[i];
        }
    }
    cout<<ans<<endl;
    #ifndef ONLINE_JUDGE
        cerr<<"Time elapsed: "<<1.0*clock()/CLOCKS_PER_SEC<<" s."<<endl;
    #endif
    return 0;
}