Codeforces Round #248 DIV1 C問題/DIV2 E問題 - Tachibana Kanade's Tofu

Source

Codeforces Round #248 DIV1 C問題 (1500pt)
Codeforces Round #248 DIV2 E問題 (2500pt)
Problem description

問題概要

$M$ 進数で $L_s$ 桁の整数 $L$ と $R_s$ 桁の整数 $R$ が与えられる.
$N$ 個の数列 $S_k$ と,その数列に対応する値 $V_k$ が与えられる.
数列の各要素は $0$ 以上 $M-1$ 以下で,$M$ 進数の数の部分文字列と思える.($S_k$ は最初が $0$ から始まることもある)
$L$ 以上 $R$ 以下の $M$ 進数の整数 $x$ であって,$x$ のスコアが $K$ 以下となるものはいくつあるかを ${\rm mod}\ 1000000007\ (10^9+7)$ で求める問題.
$x$ のスコアは,最初は $0$ で,$x$ の部分文字列として $S_k$ が現れるたびに $V_k$ が足されて,計算されるもの.
同じ $S_k$ が複数回 $x$ の部分文字列として現れるなら,その都度スコアに足される.

解法

条件を満たす $R$ 以下の整数の数から条件を満たす $L-1$ 以下の整数の数を引けば良いので,桁DPする.
状態として,(最後に使った文字列がどの文字列の最初何文字と一致しているか,今までのスコアの値)を状態して,そのようなパターン数を計算する.
一致してる文字列はすべて記憶する必要はなく,最も長く一致しているもののみを覚えておけば,状態を復元できる.

C++のスパゲッティなコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define READER_BUF_SIZE 1048576
#define WRITER_BUF_SIZE 1048576
int reader_pt=READER_BUF_SIZE,reader_last;char reader_buf[READER_BUF_SIZE];
int writer_pt=0;char writer_buf[WRITER_BUF_SIZE];
#define mygc(c) {if(reader_pt==READER_BUF_SIZE)reader_pt=0,reader_last=fread(reader_buf,sizeof(char),READER_BUF_SIZE,stdin);(c)=reader_buf[reader_pt++];}
#define mypc(c) {if(writer_pt==WRITER_BUF_SIZE)writer_pt=0,fwrite(writer_buf,sizeof(char),WRITER_BUF_SIZE,stdout);writer_buf[writer_pt++]=(c);}
#define myed {fwrite(writer_buf,sizeof(char),writer_pt,stdout);writer_pt=0;}

#define ll long long
#define MD 1000000007

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void reader(int *x, int *y, int *z){reader(x);reader(y);reader(z);}
void reader(ll *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
int reader(char c[]){int i,s=0;for(;;){mygc(i);if(i!=' '&&i!='\n'&&i!='\r'&&i!='\t'&&i!=EOF) break;}c[s++]=i;for(;;){mygc(i);if(i==' '||i=='\n'||i=='\r'||i=='\t'||i==EOF) break;c[s++]=i;}return s;}

void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}
void writer(ll x, char c){int i,sz=0,m=0;char buf[20];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}
void writer(const char c[]){int i;for(i=0;c[i]!='\0';i++)mypc(c[i]);}

int M, N, K;
int Ls, L[210];
int Rs, R[210];
int V[210], Ss[210], S[210][210];

int node;
int st1[400], st2[400];
int edge[400][210], cost[400][210];

ll dp[400][2][501], nx[400][2][501];

int solve(int n, int A[]){
  int i, j, k, d, z;
  ll res = 0;

  rep(i,node) rep(j,2) rep(k,K+1) dp[i][j][k] = 0;
  dp[0][1][0] = 1;

  rep(d,n){
    rep(i,node) rep(j,2) rep(k,K+1) nx[i][j][k] = 0;

    rep(i,node) rep(k,K+1) if(dp[i][0][k]){
      rep(z,M) if(k+cost[i][z]<=K) nx[edge[i][z]][0][k+cost[i][z]] += dp[i][0][k];
    }
    
    rep(i,node) rep(k,K+1) if(dp[i][1][k]){
      rep(z,A[d]) if(k+cost[i][z]<=K) nx[edge[i][z]][0][k+cost[i][z]] += dp[i][1][k];
      if(k+cost[i][z]<=K) nx[edge[i][z]][1][k+cost[i][z]] += dp[i][1][k];
    }
    
    rep(i,node) rep(j,2) rep(k,K+1){
      dp[i][j][k] = nx[i][j][k];
      if(dp[i][j][k] >= MD) dp[i][j][k] %= MD;
    }
  }

  rep(i,node) rep(j,2) rep(k,K+1) res += dp[i][j][k];
  res %= MD;
  return res;
}

int get_node(int s1, int s2){
  int i;
  REP(i,1,node) if(s1==st1[i] && s2==st2[i]) return i;
  st1[node] = s1;
  st2[node] = s2;
  return node++;
}

int main(){
  int i, j, k;
  int x, y, mx, my;
  int arr[210];
  ll res;

  reader(&N, &M, &K);
  reader(&Ls); rep(i,Ls) reader(L+i);
  reader(&Rs); rep(i,Rs) reader(R+i);
  rep(i,N){
    reader(Ss+i);
    rep(j,Ss[i]) reader(S[i]+j);
    reader(V+i);
  }

  node = 2;
  st1[0] = st2[0] = 0;
  st1[1] = st2[1] = 0;
  rep(k,node){
    rep(i,st2[k]) arr[i] = S[st1[k]][i];
    rep(i,M){
      
      if(i==0 && k==0){
        edge[k][i] = 0;
        cost[k][i] = 0;
        continue;
      }
      cost[k][i] = 0;
      arr[st2[k]] = i;
      mx = my = 0;
      rep(x,N){
        for(y=min(st2[k]+1, Ss[x]);y;y--){
          if(y!=Ss[x] && y<=my) continue;

          rep(j,y) if(S[x][j]!=arr[st2[k]-y+1+j]) break;
          if(j != y) continue;
          if(y==Ss[x]){ cost[k][i]+=V[x]; continue; }
          if(y > my){
            mx = x; my = y;
            break;
          }
        }
      }
      edge[k][i] = get_node(mx, my);
    }
  }

//  printf("%d\n",node);
//  rep(i,node){ rep(j,M) printf("%d -> %d (str %d, cost %d)\n",i,edge[i][j],j,cost[i][j]); }

  L[Ls-1]--;
  for(i=Ls-1;i;i--) if(L[i]<0){
    L[i] += M;
    L[i-1]--;
  }

//  printf("%d %d\n",solve(Rs, R),solve(Ls, L)); return 0;
  res = solve(Rs, R) - solve(Ls, L);
  res %= MD;
  if(res < 0) res += MD;
  writer((int)res,'\n');
  
  myed;
  return 0;
}

Current time: 2017年09月25日11時31分55秒
Last modified: 2014年05月24日20時11分16秒 (by laycrs)
Tags: Competitive_Programming Codeforces CF248 CF_Div1_C CF_Div2_E
トップページに戻る

Logged in as: unknown user (not login)

ログイン: