#261130 (C90) No.5002 stick xor

提出ソース
結果

問題	No.5002 stick xor
コンテスト
ユーザー	akakimidori
提出日時	2018-05-28 18:51:17
言語	C90 (gcc 12.3.0)
結果	WA
実行時間	-
コード長	7,075 bytes
コンパイル時間	6,665 ms
実行使用メモリ	956 KB
スコア	0
最終ジャッジ日時	2018-05-28 18:51:26
ジャッジサーバーID （参考情報）	judge9 /
このコードへのチャレンジ
（要ログイン）
ファイルパターン	結果
other	WA * 32
権限があれば一括ダウンロードができます
ソースコード

raw source code
#include<stdio.h>
#include<stdlib.h>
#include<math.h>

typedef long long int int64;

#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)<(b)?(a):(b))
#define ABS(a) ((a)>(0)?(a):-(a))

#define POS(i,j) ((i)*(n+1)+(j))

typedef struct length{
  int l;
  int index;
} len;

int cmp(const void *a,const void *b){
  return -(((len *)a)->l-((len *)b)->l);
}

void flip(int l,char *s,int n,int *t){
  int a=t[0];
  int b=t[1];
  if(a==t[2]){
    int i;
    for(i=0;i<l;i++){
      int index=POS(a,b+i);
      s[index]=s[index]=='0'?'1':'0';
    }
  } else {
    int i;
    for(i=0;i<l;i++){
      int index=POS(a+i,b);
      s[index]=s[index]=='0'?'1':'0';
    }
  }
  return;
}

int func(int l,char *s,const int n,int *a){
  int max=-l-1;
  int i,j;
  for(j=0;j<n;j++){
    int c=0;
    for(i=0;i<l;i++){
      c+=(s[POS(i,j)]=='0'?-1:1);
    }
    i=0;
    if(c>max){
      max=c;
      a[0]=i;
      a[1]=j;
      a[2]=i+l-1;
      a[3]=j;
    }
    while(i+l<n){
      c-=(s[POS(i,j)]=='0'?-1:1);
      c+=(s[POS(i+l,j)]=='0'?-1:1);
      i++;
      if(c>max){
	max=c;
	a[0]=i;
	a[1]=j;
	a[2]=i+l-1;
	a[3]=j;
      }
    }
  }
  for(i=0;i<n;i++){
    int c=0;
    for(j=0;j<l;j++){
      c+=(s[POS(i,j)]=='0'?-1:1);
    }
    j=0;
    if(c>max){
      max=c;
      a[0]=i;
      a[1]=j;
      a[2]=i;
      a[3]=j+l-1;
    }
    while(j+l<n){
      c-=(s[POS(i,j  )]=='0'?-1:1);
      c+=(s[POS(i,j+l)]=='0'?-1:1);
      j++;
      if(c>max){
	max=c;
	a[0]=i;
	a[1]=j;
	a[2]=i;
	a[3]=j+l-1;
      }
    }
  }
  flip(l,s,n,a);
  return max;
}

int score(const char *s,const int n){
  int res=0;
  int i,j;
  for(i=0;i<n;i++){
    for(j=0;j<n;j++){
      res+=(s[POS(i,j)]=='0'?1:0);
    }
  }
  return res;
}

void print(const char *s,const int n){
  int i,j;
  for(i=0;i<n;i++){
    for(j=0;j<n;j++){
      putchar(s[POS(i,j)]);
    }
    putchar('\n');
  }
  putchar('\n');
  return;
}

void swap(len *array,int x,int y){
  len t=array[x];
  array[x]=array[y];
  array[y]=t;
  return;
}

void shuffle(len *array,int len){
  int i;
  for(i=0;i<len;i++) swap(array,i,rand()%len);
  return;
}

int func_algo6(int l1,int l2,char *s,int n,int *a,int *b){
  int max=-l1-1;
  int i,j,k;
  for(i=0;i<n;i++){
    for(j=0;j+l1-1<n;j++){
      for(k=0;k+l2-1<l1;k++){
	int c=0;
	int x;
	for(x=j;x<j+k;x++) c+=(s[POS(i,x)]=='0'?-1:1);
	for(x=j+k+l2;x<j+l1;x++) c+=(s[POS(i,x)]=='0'?-1:1);
	if(c>max){
	  max=c;
	  a[0]=i;
	  a[1]=j;
	  a[2]=i;
	  a[3]=j+l1-1;
	  b[0]=i;
	  b[1]=j+k;
	  b[2]=i;
	  b[3]=j+k+l2-1;
	}
      }
      for(;k<l1 && j+k+l2-1<n;k++){
	int c=0;
	int x;
	for(x=j;x<j+k;x++) c+=(s[POS(i,x)]=='0'?-1:1);
	for(x=l1;x<k+l2;x++) c+=(s[POS(i,x)]=='0'?-1:1);
	if(c>max){
	  max=c;
	  a[0]=i;
	  a[1]=j;
	  a[2]=i;
	  a[3]=j+l1-1;
	  b[0]=i;
	  b[1]=j+k;
	  b[2]=i;
	  b[3]=j+k+l2-1;
	}
      }
    }
  }
  for(i=0;i+l1-1<n;i++){
    for(j=0;j<n;j++){
      for(k=0;k+l2-1<l1;k++){
	int c=0;
	int x;
	for(x=i;x<i+k;x++) c+=(s[POS(x,j)]=='0'?-1:1);
	for(x=i+k+l2;x<i+l1;x++) c+=(s[POS(x,j)]=='0'?-1:1);
	if(c>max){
	  max=c;
	  a[0]=i;
	  a[1]=j;
	  a[2]=i+l1-1;
	  a[3]=j;
	  b[0]=i+k;
	  b[1]=j;
	  b[2]=i+k+l2-1;
	  b[3]=j;
	}
      }
      for(;k<l1 && i+k+l2-1<n;k++){
	int c=0;
	int x;
	for(x=i;x<i+k;x++) c+=(s[POS(x,j)]=='0'?-1:1);
	for(x=l1;x<k+l2;x++) c+=(s[POS(x,j)]=='0'?-1:1);
	if(c>max){
	  max=c;
	  a[0]=i;
	  a[1]=j;
	  a[2]=i+l1-1;
	  a[3]=j;
	  b[0]=i+k;
	  b[1]=j;
	  b[2]=i+k+l2-1;
	  b[3]=j;
	}
      }
    }
  }
  flip(l1,s,n,a);
  flip(l2,s,n,b);
  return max;
}

void algo6(char *s,int n,len *array,int k,int *ans,int *tmp){
  const int parameter=4;
  int i;
  for(i=0;i<k && array[i].l>=parameter;i++);
  const int m=i/2*2;
  for(i=0;i<m;i++) func(array[i].l,s,n,ans+4*array[i].index);
  int maxScore=score(s,n);
  int iter=10;
  while(iter--){
    for(i=0;i<4*k;i++) tmp[i]=ans[i];
    for(i=0;i<m;i++) swap(array,i,rand()%m);
    for(i=0;i<m;i++) flip(array[i].l,s,n,tmp+4*array[i].index);
    for(i=0;i<m;i+=2){
      int a=func(array[i].l,s,n,tmp+4*array[i].index);
      int b=func(array[i+1].l,s,n,tmp+4*array[i+1].index);
      if(ABS(array[i].l-array[i+1].l)>=3 && (a<=0 || b<=0)){
	flip(array[i].l,s,n,tmp+4*array[i].index);
	flip(array[i+1].l,s,n,tmp+4*array[i+1].index);
	int tmp0[4];
	int tmp1[4];
	int c;
	if(array[i].l>array[i+1].l){
	  c=func_algo6(array[i].l,array[i+1].l,s,n,tmp0,tmp1);
	} else {
	  c=func_algo6(array[i+1].l,array[i].l,s,n,tmp1,tmp0);
	}
	if(c<a+b){
	  flip(array[i].l,s,n,tmp0);
	  flip(array[i+1].l,s,n,tmp1);
	  flip(array[i].l,s,n,tmp+4*array[i].index);
	  flip(array[i+1].l,s,n,tmp+4*array[i+1].index);
	} else {
	  tmp[4*array[i].index+0]=tmp0[0];
	  tmp[4*array[i].index+1]=tmp0[1];
	  tmp[4*array[i].index+2]=tmp0[2];
	  tmp[4*array[i].index+3]=tmp0[3];
	  tmp[4*array[i+1].index+0]=tmp1[0];
	  tmp[4*array[i+1].index+1]=tmp1[1];
	  tmp[4*array[i+1].index+2]=tmp1[2];
	  tmp[4*array[i+1].index+3]=tmp1[3];
	}
      }
    }
    int localScore=score(s,n);
    if(localScore>maxScore){
      maxScore=localScore;
      for(i=0;i<4*k;i++) ans[i]=tmp[i];
    } else {
      for(i=0;i<m;i++){
	flip(array[i].l,s,n,tmp+4*array[i].index);
	flip(array[i].l,s,n,ans+4*array[i].index);
      }
    }
  }
  for(i=m;i<k;i++){
    func(array[i].l,s,n,ans+4*array[i].index);
  }
  return;
}

void algo7(char *s,int n,len *array,int k,int *ans,int *tmp){
  const int parameter=4;
  int i;
  for(i=0;i<k && array[i].l>=parameter;i++);
  const int m=i;
  len *now=(len *)malloc(sizeof(len)*m);
  for(i=0;i<m;i++) now[i]=array[i];
  len *next=(len *)malloc(sizeof(len)*m);
  int nowLen=m;
  int flag=1;
  while(flag && nowLen>0){
    printf("score : %d\n",score(s,n));
    print(s,n);

    flag=0;
    shuffle(now,nowLen);
    int nextLen=0;
    for(i=0;i<m;i++){
      int t=func(now[i].l,s,n,ans+4*now[i].index);
      if(t<=0){
	flip(now[i].l,s,n,ans+4*now[i].index);
	next[nextLen++]=now[i];
      } else {
	flag=1;
      }
    }
    len *swapL=now;
    now=next;
    next=swapL;
    nowLen=nextLen;
  }
  if(nowLen%2!=0){
    func(now[nowLen-1].l,s,n,ans+4*now[nowLen-1].index);
  }
  for(i=0;i<nowLen/2*2;i+=2){
    if(now[i].l>now[i+1].l){
      func_algo6(now[i].l,now[i+1].l,s,n,ans+4*now[i].index,ans+4*now[i+1].index);
    } else {
      func_algo6(now[i+1].l,now[i].l,s,n,ans+4*now[i+1].index,ans+4*now[i].index);
    }
  }
  printf("score : %d\n",score(s,n));
  print(s,n);

  for(i=m;i<k;i++){
    func(array[i].l,s,n,ans+4*array[i].index);
  }
  printf("score : %d\n",score(s,n));
  print(s,n);
  return;
}

void run(void){
  int n,k;
  scanf("%d%d",&n,&k);
  char *s=(char *)calloc((n+1)*n,sizeof(char));
  len *array=(len *)malloc(sizeof(len)*k);
  int *ans=(int *)malloc(sizeof(int)*4*k);
  int *tmp=(int *)malloc(sizeof(int)*4*k);
  int i;
  for(i=0;i<k;i++){
    int t;
    scanf("%d",&t);
    array[i].l=t;
    array[i].index=i;
  }
  qsort(array,k,sizeof(len),cmp);
  for(i=0;i<n;i++) scanf("%s",s+POS(i,0));
  algo7(s,n,array,k,ans,tmp);
  for(i=0;i<k;i++){
    printf("%d %d %d %d\n",ans[i*4+0]+1,ans[i*4+1]+1,ans[i*4+2]+1,ans[i*4+3]+1);
  }
  return;
}

int main(void){
  run();
  return 0;
}
yukicoder

結果

ソースコード
