Gym - 100851G：Generators（人尽皆知但是WA题）

题意：现在有函数，每一项Xi=（A*X(i-1)+B）%C。现在给定N个函数以及K：X0，A，B，C。然你再每个函数选择一个数，使得其和最大，而且不被K整除。

X0，A，B，C<=1e3 ；K<=1e9

思路：因为C比较小，我们可以找到循环节（不一定是从0开始就循环了，所以用vis判定是否出现过）。然后保存每个函数的最大值，以及不同余的第二大值。

如果所有数的最大值不被K整除，那么最多有一个数被第二大替代

（一个简单题我居然wa了几发，所以还是写一下。

#include<bits/stdc++.h>
#define pii pair<int,int>
#define F first
#define S second
#define mp make_pair
using namespace std;
const int maxn=;
pii Mx[maxn][]; int pos[maxn];
int a[maxn],vis[maxn];
int main()
{
    freopen("generators.in","r",stdin);
    freopen("generators.out","w",stdout);
    int N,K,i,j,x,A,B,C,ans=,tmp,L;
    scanf("%d%d",&N,&K);
    for(i=;i<=N;i++){
        scanf("%d",&a[]); Mx[i][]=mp(a[],);
        scanf("%d%d%d",&A,&B,&C);
        vis[a[]]=; L=;
        for(j=;;j++){
            a[j]=(a[j-]*A+B)%C;
            if(vis[a[j]]) break;
            if(a[j]>Mx[i][].F) Mx[i][]=mp(a[j],j);
            vis[a[j]]=; L++;
        }
        Mx[i][]=mp(-,-);
        for(j=;j<=L;j++)
            if(a[j]%K!=Mx[i][].F%K&&a[j]>Mx[i][].F) Mx[i][]=mp(a[j],j);
        for(j=;j<=L;j++) vis[a[j]]=;
    }
    for(i=;i<=N;i++) ans+=Mx[i][].F;
    for(i=;i<=N;i++) pos[i]=Mx[i][].S;
    if(ans%K!=) printf("%d\n",ans);
    else {
        int change=,sum=-;
        for(i=;i<=N;i++){
            if(Mx[i][].F!=-){
                if(ans-Mx[i][].F+Mx[i][].F>sum) sum=ans-Mx[i][].F+Mx[i][].F,change=i;
            }
        }
        if(sum==-||sum%K==) { puts("-1"); return ; }
        printf("%d\n",sum);
        pos[change]=Mx[change][].S;
    }
    for(i=;i<=N;i++) printf("%d ",pos[i]);
    return ;
}