結果

問題 No.2709 1975 Powers
コンテスト
ユーザー tails
提出日時 2024-04-01 10:24:22
言語 C90
(gcc 12.3.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,385 bytes
コンパイル時間 599 ms
コンパイル使用メモリ 25,600 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-09-30 21:45:12
合計ジャッジ時間 1,373 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function ‘main’:
main.c:7:59: warning: implicit declaration of function ‘write’ [-Wimplicit-function-declaration]
    7 | #define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
      |                                                           ^~~~~
main.c:76:17: note: in expansion of macro ‘wt1’
   76 |                 wt1(z);
      |                 ^~~
main.c:78:9: warning: implicit declaration of function ‘_exit’ [-Wimplicit-function-declaration]
   78 |         _exit(0);
      |         ^~~~~
main.c:78:9: warning: incompatible implicit declaration of built-in function ‘_exit’ [-Wbuiltin-declaration-mismatch]

ソースコード

diff # 

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

#define rd_init() char*rp=({char*mmap();mmap(0l,1l<<25,1,2,0,0ll);})
#define rd() ({int _v=0,_c;while(_c=*rp++-48,_c>=0)_v=_v*10+_c;_v;})
#define wt(v) ({ulong _z=v;do*--wp=_z%10+48;while(_z/=10);})
#define wt1(v) ({char wbuf[64],*wp=wbuf+sizeof wbuf;wt(v);write(1,wp,wbuf+sizeof wbuf-wp);})
#define rep(v,e) for(typeof(e) v=0;v<e;++v)
#define rrep(v,e) for(typeof(e) v=e;v--;)

typedef unsigned long ulong;

ulong h[2000000/64+1];
unsigned as[200];
int d[100000];
int m10[200],m9[200],m7[200],m5[200];

int main(){
	int n;
	unsigned p,q;
	{
		rd_init();
		n=rd();
		p=rd();
		q=rd();
		rep(i,n){
			int a=rd();
			h[a>>6]|=1ul<<a;
		}
	}
	{
		int k=0;
		rep(i,2000000/64+1){
			if(h[i]){
				rep(j,64){
					if(h[i]&1ul<<j){
						as[k++]=i<<6|j;
					}
				}
			}
		}
	}
	{
		int z=0;
		rep(i,n){
			ulong a10=10,a9=9,a7=7,a5=5;
			ulong b10=1,b9=1,b7=1,b5=1;
			for(unsigned k=as[i];k;k>>=1){
				if(k&1){
					b10=b10*a10%p;
					b9=b9*a9%p;
					b7=b7*a7%p;
					b5=b5*a5%p;
				}
				a10=a10*a10%p;
				a9=a9*a9%p;
				a7=a7*a7%p;
				a5=a5*a5%p;
			}
			m10[i]=b10;
			m9[i]=b9;
			m7[i]=b7;
			m5[i]=b5;
		}
		rrep(ci,n){
			unsigned cv=m7[ci];
			rrep(bi,ci){
				unsigned bv=m9[bi];
				rrep(ai,bi){
					unsigned av=m10[ai];
					z+=d[(p*3+q-(cv+bv+av))%p];
				}
			}
			d[m5[ci]]+=1;
		}
		wt1(z);
	}
	_exit(0);
}
0