#591658 (C) No.1310 量子アニーリング

• 問題: No.1310 量子アニーリング
• ユーザー: tails
• 提出日時: 2020-12-10 17:05:03
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 3 ms / 2,000 ms
• コード長: 650 bytes
• コンパイル時間: 309 ms
• コンパイル使用メモリ: 30,720 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-19 20:38:09
• 合計ジャッジ時間: 1,212 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	3 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	3 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	3 ms; 5,376 KB
testcase_18	AC	3 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	3 ms; 5,376 KB
testcase_22	AC	2 ms; 5,376 KB
testcase_23	AC	2 ms; 5,376 KB

コンパイルメッセージ

main.c:33:1: warning: return type defaults to 'int' [-Wimplicit-int]
   33 | main(){
      | ^~~~
main.c: In function 'main':
main.c:36:9: warning: implicit declaration of function 'scanf' [-Wimplicit-function-declaration]
   36 |         scanf("%d",&n);
      |         ^~~~~
main.c:1:1: note: include '<stdio.h>' or provide a declaration of 'scanf'
  +++ |+#include <stdio.h>
    1 | #pragma GCC optimize("Ofast")
main.c:36:9: warning: incompatible implicit declaration of built-in function 'scanf' [-Wbuiltin-declaration-mismatch]
   36 |         scanf("%d",&n);
      |         ^~~~~
main.c:36:9: note: include '<stdio.h>' or provide a declaration of 'scanf'
main.c:47:9: warning: implicit declaration of function 'printf' [-Wimplicit-function-declaration]
   47 |         printf("%d",(y+z)*2%MD*inverse(ifac[n])%MD);
      |         ^~~~~~
main.c:47:9: note: include '<stdio.h>' or provide a declaration of 'printf'
main.c:47:9: warning: incompatible implicit declaration of built-in function 'printf' [-Wbuiltin-declaration-mismatch]
main.c:47:9: note: include '<stdio.h>' or provide a declaration of 'printf'

ソースコード

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

#define MD 998244353

int inverse(int a){
	int b=MD;
	int u=1;
	int v=0;
	int s,t;
	while(b){
		t=a/b;
		s=b; b=a-t*b; a=s;
		s=v; v=u-t*v; u=s;
	}
	if(u<0){
		u+=MD;
	}
	return u;
}

#define M 200001

int ifac[M];

void mkfac(){
	long x=571834101;
	for(int i=M;i--;){
		ifac[i]=x=x*(i+1)%MD;
	}
}

main(){
	mkfac();
	int n,h;
	scanf("%d",&n);
	h=(n>>1)+1;
	long y=0,z=0;
	for(int i=0;i<h;i+=2){
		y=(y*16+(long)ifac[i]*(long)ifac[n-i])%MD;
	}
	y<<=n&3;
	for(int i=n&~1;i>=h;i-=2){
		z=(z*16+(long)ifac[i]*(long)ifac[n-i])%MD;
	}
	z<<=4-(n&3);
	printf("%d",(y+z)*2%MD*inverse(ifac[n])%MD);
}