#1031266 (C) No.2989 Fibonacci Prize

提出ソース

結果

問題	No.2989 Fibonacci Prize
ユーザー	ygussany
提出日時	2024-12-14 01:15:54
言語	C (gcc 12.3.0)
結果	WA
実行時間	-
コード長	3,376 bytes
コンパイル時間	457 ms
コンパイル使用メモリ	34,048 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-12-14 01:15:58
合計ジャッジ時間	2,904 ms
ジャッジサーバーID （参考情報）	judge5 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	1 ms 5,248 KB
testcase_02	AC	4 ms 5,248 KB
testcase_03	AC	1 ms 5,248 KB
testcase_04	AC	1 ms 5,248 KB
testcase_05	AC	1 ms 5,248 KB
testcase_06	AC	1 ms 5,248 KB
testcase_07	AC	1 ms 5,248 KB
testcase_08	AC	1 ms 5,248 KB
testcase_09	AC	1 ms 5,248 KB
testcase_10	AC	1 ms 5,248 KB
testcase_11	AC	1 ms 5,248 KB
testcase_12	AC	1 ms 5,248 KB
testcase_13	AC	1 ms 5,248 KB
testcase_14	AC	1 ms 5,248 KB
testcase_15	AC	1 ms 5,248 KB
testcase_16	AC	1 ms 5,248 KB
testcase_17	AC	1 ms 5,248 KB
testcase_18	AC	1 ms 5,248 KB
testcase_19	AC	1 ms 5,248 KB
testcase_20	AC	1 ms 5,248 KB
testcase_21	AC	1 ms 5,248 KB
testcase_22	AC	1 ms 5,248 KB
testcase_23	AC	2 ms 5,248 KB
testcase_24	AC	3 ms 5,248 KB
testcase_25	AC	5 ms 5,248 KB
testcase_26	AC	1 ms 5,248 KB
testcase_27	AC	3 ms 5,248 KB
testcase_28	AC	2 ms 5,248 KB
testcase_29	AC	1 ms 5,248 KB
testcase_30	AC	1 ms 5,248 KB
testcase_31	AC	1 ms 5,248 KB
testcase_32	AC	1 ms 5,248 KB
testcase_33	AC	4 ms 5,248 KB
testcase_34	AC	2 ms 5,248 KB
testcase_35	AC	2 ms 5,248 KB
testcase_36	AC	5 ms 5,248 KB
testcase_37	AC	5 ms 5,248 KB
testcase_38	AC	2 ms 5,248 KB
testcase_39	AC	3 ms 5,248 KB
testcase_40	AC	2 ms 5,248 KB
testcase_41	AC	3 ms 5,248 KB
testcase_42	AC	5 ms 5,248 KB
testcase_43	AC	12 ms 5,248 KB
testcase_44	AC	14 ms 5,248 KB
testcase_45	AC	13 ms 5,248 KB
testcase_46	AC	9 ms 5,248 KB
testcase_47	AC	11 ms 5,248 KB
testcase_48	AC	10 ms 5,248 KB
testcase_49	AC	10 ms 5,248 KB
testcase_50	AC	10 ms 5,248 KB
testcase_51	AC	9 ms 5,248 KB
testcase_52	AC	9 ms 5,248 KB
testcase_53	AC	10 ms 5,248 KB
testcase_54	AC	10 ms 5,248 KB
testcase_55	AC	9 ms 5,248 KB
testcase_56	AC	8 ms 5,248 KB
testcase_57	AC	8 ms 5,248 KB
testcase_58	AC	10 ms 5,248 KB
testcase_59	AC	9 ms 5,248 KB
testcase_60	AC	10 ms 5,248 KB
testcase_61	AC	8 ms 5,248 KB
testcase_62	AC	8 ms 5,248 KB
testcase_63	AC	8 ms 5,248 KB
testcase_64	WA	-
testcase_65	AC	1 ms 5,248 KB
testcase_66	AC	1 ms 5,248 KB
testcase_67	AC	1 ms 5,248 KB
testcase_68	AC	1 ms 5,248 KB
testcase_69	AC	1 ms 5,248 KB
testcase_70	AC	1 ms 5,248 KB
testcase_71	AC	1 ms 5,248 KB
testcase_72	AC	1 ms 5,248 KB
testcase_73	AC	1 ms 5,248 KB
testcase_74	AC	1 ms 5,248 KB
testcase_75	AC	1 ms 5,248 KB
testcase_76	AC	1 ms 5,248 KB
testcase_77	AC	1 ms 5,248 KB
testcase_78	AC	1 ms 5,248 KB
testcase_79	AC	1 ms 5,248 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

main.c: In function 'solve':
main.c:65:32: warning: implicit declaration of function 'exit' [-Wimplicit-function-declaration]
   65 |                 if (i > M - 2) exit(-1);
      |                                ^~~~
main.c:2:1: note: include '<stdlib.h>' or provide a declaration of 'exit'
    1 | #include <stdio.h>
  +++ |+#include <stdlib.h>
    2 | 
main.c:65:32: warning: incompatible implicit declaration of built-in function 'exit' [-Wbuiltin-declaration-mismatch]
   65 |                 if (i > M - 2) exit(-1);
      |                                ^~~~
main.c:65:32: note: include '<stdlib.h>' or provide a declaration of 'exit'

ソースコード

raw source code

#include <stdio.h>

int naive(int N, int M)
{
	int i;
	static __int128 F[1001];
	for (i = 3, F[1] = 1, F[2] = 1; i <= M; i++) F[i] = F[i-1] + F[i-2];
	
	int ans = 0, l, r;
	__int128 sum;
	for (l = 1; l <= M; l++) {
		for (r = l; r <= M; r++) {
			for (i = l, sum = 0; i <= r; i++) {
				sum += F[i];
				if (sum > F[M]) break;
			}
			if (i > r && sum % N == 0) ans++;
		}
	}
	return ans;
}

long long solve(int N, int M)
{
	if (M == 1) {
		if (N == 1) return 1;
		else return 0;
	} else if (M == 2) {
		if (N == 1) return 2;
		else return 0;
	} else if (M == 3) {
		if (N == 1) return 4;
		else if (N == 2) return 2;
		else return 0;
	} else if (N == 1) return (long long)(M - 2) * (M - 1) / 2 + 3;
	
	int i, cur, prev, F[2], sum = 2 % N, flag = 0;
	static int num[200001], tmp[200001];
	for (i = 0; i < N; i++) {
		num[i] = 0;
		tmp[i] = 0;
	}
	num[0]++;
	num[1]++;
	num[sum]++;
	for (i = 3, F[0] = 1, F[1] = 1, cur = 1, prev = 0; i <= M - 2; i++, cur ^= 1, prev ^= 1) {
		F[prev] += F[cur];
		if (F[prev] >= N) F[prev] -= N;
		if (sum == 0 && F[prev] == 0 && F[cur] == 1) {
			if (flag == 0) flag = i;
			else {
				flag = i - flag;
				break;
			}
		}
		sum += F[prev];
		if (sum >= N) sum -= N;
		num[sum]++;
		if (flag > 0) tmp[sum]++;
	}
	if (i <= M - 2) {
		int j, k = (M - 2 - i) / flag;
		for (j = 0; j < N; j++) num[j] += k * tmp[j];
		i += k * flag;
		if (i > M - 2) exit(-1);
		for (num[sum]++, i++, cur ^= 1, prev ^= 1; i <= M - 2; i++, cur ^= 1, prev ^= 1) {
			F[prev] += F[cur];
			if (F[prev] >= N) F[prev] -= N;
			sum += F[prev];
			if (sum >= N) sum -= N;
			num[sum]++;
		}
	}
	
	long long ans = 0;
	for (i = 0; i < N; i++) ans += (long long)num[i] * (num[i] - 1) / 2;
	F[prev] += F[cur];
	if (F[prev] % N == 0) ans++;
	if ((F[prev] + F[cur]) % N == 0) ans += 2;
	return ans;
}

#define MT_N 624
#define MT_M 397
#define MT_MATRIX_A 0x9908b0dfUL
#define MT_UPPER_MASK 0x80000000UL
#define MT_LOWER_MASK 0x7fffffffUL

static unsigned int mt[MT_N];
static int mti = MT_N + 1;

void init_genrand(unsigned int s)
{
    mt[0] = s & 0xffffffffUL;
    for (mti = 1; mti < MT_N; mti++) {
        mt[mti] = (1812433253UL * (mt[mti-1] ^ (mt[mti-1] >> 30)) + mti); 
        mt[mti] &= 0xffffffffUL;
    }
}

unsigned int genrand()
{
    unsigned int y;
    static unsigned int mag01[2] = {0x0UL, MT_MATRIX_A};

    if (mti >= MT_N) {
        int kk;
        if (mti == MT_N + 1) init_genrand(5489UL);
		
        for (kk = 0; kk < MT_N - MT_M; kk++) {
            y = (mt[kk] & MT_UPPER_MASK) | (mt[kk+1] & MT_LOWER_MASK);
            mt[kk] = mt[kk+MT_M] ^ (y >> 1) ^ mag01[y&0x1UL];
        }
        for (; kk < MT_N - 1; kk++) {
            y = (mt[kk] & MT_UPPER_MASK) | (mt[kk+1] & MT_LOWER_MASK);
            mt[kk] = mt[kk+(MT_M-MT_N)] ^ (y >> 1) ^ mag01[y&0x1UL];
        }
        y = (mt[MT_N-1] & MT_UPPER_MASK) | (mt[0] & MT_LOWER_MASK);
        mt[MT_N-1] = mt[MT_M-1] ^ (y >> 1) ^ mag01[y&0x1UL];

        mti = 0;
    }
  
    y = mt[mti++];

    y ^= (y >> 11);
    y ^= (y << 7) & 0x9d2c5680UL;
    y ^= (y << 15) & 0xefc60000UL;
    y ^= (y >> 18);

    return y;
}

int main()
{
	int N, M;
	scanf("%d %d", &N, &M);
	printf("%lld\n", solve(N, M));
	
	/*
	for (N = 1; N <= 30; N++) {
		for (M = 1; M <= 30; M++) {
			if (solve(N, M) != naive(N, M)) printf("%d %d %lld %d\n", N, M, solve(N, M), naive(N, M));
		}
	}
	*/
	fflush(stdout);
	return 0;
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード