結果

問題 No.1536 仕切り直し
コンテスト
ユーザー tails
提出日時 2021-06-07 16:37:50
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 951 bytes
コンパイル時間 328 ms
コンパイル使用メモリ 33,896 KB
実行使用メモリ 63,380 KB
最終ジャッジ日時 2024-11-25 06:36:57
合計ジャッジ時間 1,162 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c:16:1: warning: return type defaults to 'int' [-Wimplicit-int]
   16 | main(){
      | ^~~~
main.c: In function 'main':
main.c:49:9: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   49 |         write(1,wp,wbuf+sizeof wbuf-wp);
      |         ^~~~~
main.c:50:9: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   50 |         _exit(0);
      |         ^~~~~
      |         _Exit

ソースコード

diff # 

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

char*mmap();
#define rd(v) long v=0;{int _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}
#define wt(v) {long _z=v;do*--wp=_z%10+48;while(_z/=10);}

char wbuf[1<<20];

#define min(a,b) (a<=b?a:b)
#define max(a,b) (a>=b?a:b)

int a[2001];
long gmin[2001][2001],gmax[2001][2001];

main(){
	char*rp=mmap(0l,1l<<20,1,2,0,0ll);
	rd(n);
	rd(m);
	for(int i=0;i<n;++i){
		int neg=*rp=='-'&&++rp;
		rd(ai);
		a[i]=ai=neg?-ai:ai;
		gmin[i+1][0]=gmin[i][0]+ai;
		gmax[i+1][0]=gmax[i][0]+ai;
		for(int j=1;j<=m;++j){
			gmin[i+1][j]=min(gmin[i][j]+ai,-(gmax[i][j-1]+ai));
			gmax[i+1][j]=max(gmax[i][j]+ai,-(gmin[i][j-1]+ai));
		}
	}

	char*wp=wbuf+sizeof wbuf;
	for(int i=n;i--;){
		long ai=a[i];
		if(m&1 ?
			gmin[i+1][m]!=gmin[i][m]+ai :
			gmax[i+1][m]!=gmax[i][m]+ai
		){
			*--wp='\n';
			wt(i+1);
			--m;
		}
	}
	while(m){
		*--wp='\n';
		*--wp='0';
		--m;
	}
	write(1,wp,wbuf+sizeof wbuf-wp);
	_exit(0);
}
0