ei1333の日記

ぺこい

Codeforces Round #358 (Div. 2)

ABCDを解いた.

A. Alyona and Numbers

問題概要

整数  {n, m (1 \le n, m \le 1 000 000)} が与えられる.
 {1 \le x \le n} を満たす  {x} と,  {1 \le y \le n} を満たす  {y} で,  {(x + y) \% 5 = 0} となるような ペア  {(x, y)}を作りたい.
 {(x, y)}候補数を求めよ.

解法

簡単なのは分かるけど, ちょっと面倒.

 {(x + y) \% 5 = 0} は,  {(x \% 5 + y \% 5) \% 5 = 0} としても同じである.
よって, 5 で割った余りが同じものは, まとめてしまって良い.
5で割った余りのそれぞれの個数が分かれば良いので, 適当に切り上げたりしならがら計算する.

ソース

#include <bits/stdc++.h>
using namespace std;
typedef long long int64;
int main()
{
  int64 N, M;
  cin >> N >> M;
  int64 ret = 0;
  for(int64 i = 0; i <= 4; i++) {
    int64 count1 = (N + 4 - i) / 5;
    int64 count2 = (M + (i + 1) % 5) / 5;
    ret += count1 * count2;
  }
  cout << ret << endl;
}

B. Alyona and Mex

問題概要

長さ  {n (1 \le n \le 100 000)} の数列  {a_1, a_2, .., a_n} が与えられる.
この数列のある要素  {i} を選んで, 要素の中身を  {x (1 \le x \le a_i)} に置換する操作を好きなだけ行うことが出来る.
数列に現れない正整数の最小値  {mex} を定義する. (例えば,  {1, 3, 4} なら  {mex = 2}.)
このとき  {mex} の最大値を求めよ.

解法

 {1, 2, 3, ....} のように順番に作っていくことを考える.
同じ数値を作るなら小さい数を使った方が良いので, ソートしたほうがよさそう[1]. ソートする.
このとき以下の貪欲が成立.
1 番目の要素から順に見ていく. 今  {i} 番目の要素を見ているとする.
数値  {k} まで作れているとき,  {a_i} を使って  {k + 1} を作れるかどうかは単純な大小関係. 作れたら作ってしまって良い([1] より).

ソース

#include <bits/stdc++.h>
using namespace std;
int main()
{
  int N, A[100000];
  scanf("%d", &N);
  for(int i = 0; i < N; i++) scanf("%d", &A[i]);
  sort(A, A + N);
  
  int ret = 0;
  for(int i = 0; i < N; i++) {
    if(ret < A[i]) ++ret;
  }
  printf("%d\n", ret + 1);
}

C. Alyona and the Tree

問題概要

頂点 1 を根とする根付き木が与えられる.
すべての辺, 頂点には, それぞれある値を持つ. ここで, 頂点  {i} が持つ値を  {a_i} とする.
適切に, この木から悲しい頂点がなくなるように葉を消すことを繰り返した時の, 消す葉の数の最小値を求めよ.
このとき悲しい頂点を以下のように定義する.

  •  {v} が悲しい頂点  {:=}  {v} の部分木のすべての頂点  {u_i} で,  {dist(v, u_i) \gt a_i} を満たす頂点  {u_j} が存在する.
  •  {dist(v, u) := } 頂点  {v} から頂点  {u} への経路中の辺に書かれた数の総和.

解法

まず, 部分木が云々という文字列があるのでDPっぽくできそう. 実際できる.

  •  {dp_{idx} :} 頂点  {idx} の祖先ノードから  {idx} までの経路のうちの辺の最大の和

とするとよくて,  {dp_{idx}} の親ノードの  {idx} {p_{idx}}, その間の辺のコストを  {c}とすると,  {dp_{idx} = \max(0, dp_{par_{idx}} + c)} とすれば更新できる(経路はそこの辺で切るか, 親へ行くかのどちらかなので).

最後に, 根から下っていったときに, それぞれ悲しいノードが出たところで切れば良い.

ソース

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

int64 N, A[100000];
pair< int64, int64 > par[100000];
vector< int64 > graph[100000];
int64 updp[100000];
bool memo[100000];

int64 rec(int idx)
{
  if(idx == -1)   return(0);
  if(memo[idx]++) return(updp[idx]);
  return(updp[idx] = max(0LL, rec(par[idx].first) + par[idx].second));
}
int dfs(int idx)
{
  if(updp[idx] > A[idx]) return(0);
  int ret = 1;
  for(int to : graph[idx]) ret += dfs(to);
  return(ret);
}
int main()
{
  scanf("%d", &N);
  for(int i = 0; i < N; i++) {
    scanf("%d", A + i);
  }
  par[0] = {-1, 0};
  for(int i = 1; i < N; i++) {
    int p, c;
    scanf("%d %d", &p, &c);
    --p;
    par[i] = {p, c};
    graph[p].push_back(i);
  }
  for(int i = 0; i < N; i++) rec(i);
  printf("%d\n", N - dfs(0));
}

D. Alyona and Strings

問題概要

アルファベット小文字のみで構成された文字列  {s, t (1 \le |s|, |t|, \le 1000)} がある.
 {s, t} から部分文字列を互いに素となるように  {k(1 \le k \le 10)} 個順番に取り出し並べる. この文字列を  {x, y} とする.
 {x = y} となるように取り出すとき,  {|x,y|} の値を最大化せよ.

解法

DPをすれば良い. 以下のように dpテーブルを定義する.

  •  {dp_{x, y, z}: s} {x} 文字目かつ  {t} {y} 文字目以降の文字列で,  {z} 個部分文字列を 取り出すときの最長の長さ

この dpテーブルを更新する
まず, その文字は取り出さないことを考えると, 次の文字に進める. これは明らかに以下.

  •  {dp_{x, y, z} = \max(dp_{x + 1, y, z}, dp_{x, y + 1, z})}

この文字から始まる部分文字列を取り出すことを考えると, 最大の長さの文字列を取り出す方が良い. これが何文字かは, その位置の  {LCP}.  {LCP} を求めたくなるが, これは事前に DP で求めておけば良い. この部分文字列を何個に分けるかすべての通りを試す.

  •  {dp_{x, y, z} = \sum_{i = 1}^{\min(LCP, z)} \max(dp_{x + LCP, y + LCP, z - i})}

計算量は,  {O(|s||t|k)} となって間に合う,

ソース

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

int N, M, K;
string S, T;
int LCP[1006][1006];
int dp[1006][1006][10];

int rec(int S_begin, int T_begin, int dec)
{
  if(dec == -1) return(0);
  if(S_begin >= S.size() || T_begin >= T.size()) return(-11451419);
  if(~dp[S_begin][T_begin][dec]) return(dp[S_begin][T_begin][dec]);
  int ret = max(rec(S_begin + 1, T_begin, dec), rec(S_begin, T_begin + 1, dec));
  for(int i = 1; i <= LCP[S_begin][T_begin] && dec - i >= -1; i++) {
    ret = max(ret, rec(S_begin + LCP[S_begin][T_begin], T_begin + LCP[S_begin][T_begin], dec - i) + LCP[S_begin][T_begin]);
  }
  return(dp[S_begin][T_begin][dec] = ret);
}

int main()
{
  memset(dp, -1, sizeof(dp));
  cin >> N >> M >> K; 
  cin >> S >> T;
  for(int i = S.size() - 1; i >= 0; i--) {
    for(int j = T.size() - 1; j >= 0; j--) {
      LCP[i][j] = S[i] == T[j] ? LCP[i + 1][j + 1] + 1 : 0;
    }
  }
  cout << rec(0, 0, K - 1) << endl;
}