結果

問題 No.319 happy b1rthday 2 me
ユーザー holeguma
提出日時 2015-12-13 01:18:41
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 4,218 bytes
コンパイル時間 2,587 ms
コンパイル使用メモリ 79,260 KB
実行使用メモリ 51,984 KB
最終ジャッジ日時 2024-09-15 10:58:42
合計ジャッジ時間 5,612 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other AC * 1 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.InputStream;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.InputMismatchException;
import java.util.NoSuchElementException;
import java.math.BigInteger;

public class Main{

static PrintWriter out;
static InputReader ir;

static void solve(){
 char[] a=minusOne(ir.next().toCharArray());
 char[] b=ir.next().toCharArray();
 out.println(calc(b)-calc(a)-((a[a.length-1]=='1'&&(a.length==1||a[0]=='2'))?1:0));
}

public static long calc(char[] s){
 long ret=0;
 long[][][][][] dp=new long[2][2][2][2][2];
 dp[0][0][0][0][0]=1;
 for(int i=0;i<s.length;i++){
  int now=i%2,next=now^1,num=s[i]-'0';
  for(int j=0;j<2;j++){
   for(int ft=0;ft<2;ft++){
    for(int fi=0;fi<2;fi++){
     for(int po=0;po<2;po++){
      for(int k=0;k<=(j==0?num:9);k++){
       int next_j=j|((k==num)?0:1),next_ft=ft|((fi==0&&k==2)?1:0),next_fi=fi|((k==0)?0:1),next_po=((k==1)?1:0);
       if(po==1&&k==2){
        if(next_j==0){
         long a=0;
         for(int m=i+1;m<s.length;m++){
          a*=10;
          a+=(long)(s[m]-'0');
         }
         ret+=a+1;
        }
        else ret+=(long)Math.pow(10,s.length-1-i);
       }
       dp[next][next_j][next_ft][next_fi][next_po]+=dp[now][j][ft][fi][po];
      }
     }
    }
   }
  }
  for(int j=0;j<2;j++){
   for(int k=0;k<2;k++){
    for(int l=0;l<2;l++){
     Arrays.fill(dp[now][j][k][l],0);
    }
   }
  }
 }
 ret+=dp[s.length%2][1][1][1][1];
 if(s.length>=2||s[0]>='2') ret++;
 return ret;
}

public static char[] minusOne(char[] s){
 char[] ret=new char[s.length];
 boolean f=false;
 for(int i=s.length-1;i>=0;i--){
  if(!f){
   if(s[i]!='0'){
    ret[i]=(char)(s[i]-1);
    f=true;
   }
   else ret[i]='9';
  }
  else ret[i]=s[i];
 }
 return ret;
}

public static void main(String[] args) throws Exception{
 ir=new InputReader(System.in);
 out=new PrintWriter(System.out);
 solve();
 out.flush();
}

static class InputReader {
 private InputStream in;
 private byte[] buffer=new byte[1024];
 private int curbuf;
 private int lenbuf;

 public InputReader(InputStream in) {this.in=in; this.curbuf=this.lenbuf=0;}
 
 public boolean hasNextByte() {
  if(curbuf>=lenbuf){
   curbuf= 0;
   try{
    lenbuf=in.read(buffer);
   }catch(IOException e) {
    throw new InputMismatchException();
   }
   if(lenbuf<=0) return false;
  }
  return true;
 }

 private int readByte(){if(hasNextByte()) return buffer[curbuf++]; else return -1;}
 
 private boolean isSpaceChar(int c){return !(c>=33&&c<=126);}
 
 private void skip(){while(hasNextByte()&&isSpaceChar(buffer[curbuf])) curbuf++;}
 
 public boolean hasNext(){skip(); return hasNextByte();}
 
 public String next(){
  if(!hasNext()) throw new NoSuchElementException();
  StringBuilder sb=new StringBuilder();
  int b=readByte();
  while(!isSpaceChar(b)){
   sb.appendCodePoint(b);
   b=readByte();
  }
  return sb.toString();
 }
 
 public int nextInt() {
  if(!hasNext()) throw new NoSuchElementException();
  int c=readByte();
  while (isSpaceChar(c)) c=readByte();
  boolean minus=false;
  if (c=='-') {
   minus=true;
   c=readByte();
  }
  int res=0;
  do{
   if(c<'0'||c>'9') throw new InputMismatchException();
   res=res*10+c-'0';
   c=readByte();
  }while(!isSpaceChar(c));
  return (minus)?-res:res;
 }
 
 public long nextLong() {
  if(!hasNext()) throw new NoSuchElementException();
  int c=readByte();
  while (isSpaceChar(c)) c=readByte();
  boolean minus=false;
  if (c=='-') {
   minus=true;
   c=readByte();
  }
  long res = 0;
  do{
   if(c<'0'||c>'9') throw new InputMismatchException();
   res=res*10+c-'0';
   c=readByte();
  }while(!isSpaceChar(c));
  return (minus)?-res:res;
 }

 public double nextDouble(){return Double.parseDouble(next());}

 public BigInteger nextBigInteger(){return new BigInteger(next());}

 public int[] nextIntArray(int n){
  int[] a=new int[n];
  for(int i=0;i<n;i++) a[i]=nextInt();
  return a;
 }

 public long[] nextLongArray(int n){
  long[] a=new long[n];
  for(int i=0;i<n;i++) a[i]=nextLong();
  return a;
 }

 public char[][] nextCharMap(int n,int m){
  char[][] map=new char[n][m];
  for(int i=0;i<n;i++) map[i]=next().toCharArray();
  return map;
 }
}
}
0