HDOJ 5344 MZL's xor 【math】

HDOJ 5344 MZL’s xor 【math】

題目鏈接 http://acm.hdu.edu.cn/showproblem.php?pid=5344

---

先按照規則生成數組
之後是ai 與 aj 相加的結果依次異或
建立求和矩陣
相同的數異或結果爲0 最後只保留對角線上的數
即數組中的數的兩倍
注意運算時乘法的時候可能會溢出 使用 long long

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int T, n, m, z, l;
const int MAXN = 500005;
int a[MAXN];
ll sum;

inline ll fun(int ai){
    return (ll)((ll)ai*m+z) % l;
}

void Input(){
    memset(a, 0, sizeof(a));
    scanf("%d%d%d%d", &n, &m, &z, &l);
    for(int i = 1; i < n; i++){
        a[i] = fun(a[i-1]);
    }
    for(int i = 1; i < n; i++){
        a[i] = a[i]*2;
    }
//    for(int i = 0; i < n; i++){
//        printf("%d\t", a[i]);
//    }
}

void Cale(){
    sum = 0;
    for(int i = 0; i < n; i++){
        sum = sum ^ a[i];
    }
    printf("%I64d\n", sum);
}

int main(){
    scanf("%d", &T);
    while(T--){
        Input();
        Cale();
    }

    return 0;
}

標程

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#define rep(i,j,k) for(int i=(int)j;i<=(int)k;i++)
#define per(i,j,k) for(int i=(int)j;i>=(int)k;i--)
using namespace std;
typedef long long LL;
typedef double db;
int main(){
    int T;scanf("%d",&T);
    while(T--){
        int ans=0;
        int pre=0;int m,n,z,l;
        scanf("%d%d%d%d",&n,&m,&z,&l);
        rep(i,2,n){
            pre=(pre*1ll*m+z)%l; // *1ll 轉long long 防止溢出
            ans^=pre;
        }
        printf("%d\n",ans<<1); // ai*2 == b(ai)+0
        // 0 ^ 0 == 0
        // 故 直接 b(ans)+0 即可
    }
    return 0;
}