RUPC2016 Day2 G Max Pig Noodle
問題
http://judge.u-aizu.ac.jp/onlinejudge/cdescription.jsp?cid=RitsCamp16Day2&pid=G
誤読というか,交換人と交換しなくていいパターンあるの読み取れなくて苦戦したが,自力で解けたので良かった。
考察
当初,N人の交換人との交換をどの順番に行っていくかを考えていたけど,順番を決めていくと計算量がどう考えても間に合わなそうだったので考え直した。
そこで,順番を考えずにi番目の交換人と「交換する(方法1,方法2)」「しない」をやっていき,N番目の交換人との操作まで終わって最終的に「うまか棒の数」と「ふがしの数」が0未満の数になっていなければその状態はあり得ると考えられるので,その状態での「ブタメソの数」と今までの「ブタメソの数」の最大値とのmaxを取る。
これをそれぞれの状態をメモ化しながら全パターン試していく。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.NoSuchElementException; public class Main implements Runnable{ int N,X,Y; int[] a,b,c,d; int[][][] dp; public int dfs(int n,int x,int y){ if(n >= N){ if(x < 300 || y < 300)return -1; return 0; } if(dp[n][x][y] != Integer.MIN_VALUE){ return dp[n][x][y]; } int ret = -1; { int tmp = dfs(n + 1,x - a[n],y + b[n]); if(tmp != -1){ ret = Math.max(ret,tmp); } } { int tmp = dfs(n + 1,x,y - c[n]); if(tmp != -1){ ret = Math.max(ret,tmp + d[n]); } } { int tmp = dfs(n + 1,x,y); if(tmp != -1){ ret = Math.max(ret,tmp); } } return dp[n][x][y] = ret; } public void solve() { N = nextInt(); X = nextInt(); Y = nextInt(); a = new int[N]; b = new int[N]; c = new int[N]; d = new int[N]; for(int i = 0;i < N;i++){ a[i] = nextInt(); b[i] = nextInt(); c[i] = nextInt(); d[i] = nextInt(); } dp = new int[N][901][901]; for(int i = 0;i < N;i++){ for(int j = 0;j < 901;j++){ for(int k = 0;k < 901;k++){ dp[i][j][k] = Integer.MIN_VALUE; } } } int ans = dfs(0,X + 300,Y + 300); out.println(ans == -1 ? 0 : ans); } public static void main(String[] args) { new Thread(null,new Main(),"",32 * 1024 * 1024).start(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } @Override public void run() { out.flush(); new Main().solve(); out.close(); } }
RUPC2016 Day2 D Courage Test
考察
まず,頂点uと頂点vから同じ回数しか移動できないので,Nが奇数の場合はNoになる。
次は,頂点uと頂点vそれぞれから各頂点の最短距離を記憶しておく。ダイクストラで求めても良いが,辺のコストは1なので普通にQueueを使った方が早い。
ここで,頂点u と頂点vそれぞれから最短距離がN/2の頂点を覚えておく。(実は最短距離がN/2になる頂点数は少ない)
頂点uまたは頂点vどちらかの頂点から最短距離がN/2の頂点が存在しない場合はNoとなる。
そして,(頂点uから最短距離がN/2の頂点)×(頂点vから最短距離がN/2の頂点)の組み合わせを全部試して,それぞれの最短距離の経路が被らないものがあればYesとなる。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.ArrayDeque; import java.util.ArrayList; import java.util.Arrays; import java.util.Collections; import java.util.NoSuchElementException; public class Main{ static final int INF = (int)1e9 + 7; int N,u,v; ArrayList<Integer>[] G; int[] prev1,prev2; public void dijkstra(int s,int[] dis,int[] prev){ Arrays.fill(prev, -1); Arrays.fill(dis, INF); dis[s] = 1; ArrayDeque<int[]> q = new ArrayDeque<int[]>(); q.add(new int[]{s,1}); while(q.size() > 0){ int[] p = q.poll(); if(dis[p[0]] < p[1])continue; for(int n : G[p[0]]){ if(dis[p[0]]+1 <= N/2 && dis[n] > dis[p[0]]+1){ dis[n] = dis[p[0]]+1; q.add(new int[]{n,p[1]+1}); prev[n] = p[0]; } } } } public ArrayList<Integer> getPath(int t,int[] prev){ ArrayList<Integer> path = new ArrayList<Integer>(); for(;t != -1;t = prev[t])path.add(t); Collections.reverse(path); return path; } @SuppressWarnings("unchecked") public void solve() { N = nextInt(); u = nextInt()-1; v = nextInt()-1; G = new ArrayList[N]; for(int i = 0;i < N;i++)G[i] = new ArrayList<Integer>(); for(int i = 0;i < N-1;i++){ int a = nextInt()-1; int b = nextInt()-1; G[a].add(b); G[b].add(a); } if(N % 2 == 1){ out.println("No"); return; } int[] dis1 = new int[N],dis2 = new int[N]; prev1 = new int[N]; prev2 = new int[N]; dijkstra(u,dis1,prev1); dijkstra(v,dis2,prev2); ArrayList<Integer> du = new ArrayList<Integer>(),dv = new ArrayList<Integer>(); for(int i = 0;i < N;i++){ if(dis1[i] == N/2)du.add(i); if(dis2[i] == N/2)dv.add(i); } if(du.size() == 0 || dv.size() == 0){ out.println("No"); return; } boolean[] used = new boolean[N]; for(int i = 0;i < du.size();i++){ for(int j = 0;j < dv.size();j++){ Arrays.fill(used, false); for(int n : getPath(du.get(i),prev1)){ used[n] = true; } boolean flag = true; for(int n : getPath(dv.get(j),prev2)){ if(used[n]){ flag = false; break; } } if(flag){ out.println("Yes"); return; } } } out.println("No"); } public static void main(String[] args) { out.flush(); new Main().solve(); out.close(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } }
RUPC2016 Day2 C ABNN is 17 years old forever
問題
http://judge.u-aizu.ac.jp/onlinejudge/cdescription.jsp?cid=RitsCamp16Day2&pid=C
当時は結構苦戦してたけど,今やるとUnionFindの練習ぐらいに感じる...
考察
村の合併をUnionFindで表現して,合併するごとにrankを増加させていき,最終的にrankが1の時は村とし,それ以外だったら町とする。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.Arrays; import java.util.NoSuchElementException; public class Main{ int N,M; int[] union,rank; boolean[] used; public boolean same(int x,int y){ return find(x) == find(y); } public int find(int x){ if(union[x] == x)return x; return union[x] = find(union[x]); } public void unite(int x,int y){ x = find(x); y = find(y); if(x == y)return; if(x < y){ union[y] = x; rank[x] += rank[y]; }else{ union[x] = y; rank[y] += rank[x]; } } public void solve() { N = nextInt(); M = nextInt(); union = new int[N]; rank = new int[N]; used = new boolean[N]; for(int i = 0;i < N;i++)union[i] = i; Arrays.fill(rank, 1); for(int i = 0;i < M;i++){ int a = nextInt()-1; int b = nextInt()-1; unite(a,b); } int big = 0; int small = 0; for(int i = 0;i < N;i++){ int root = find(i); if(used[root])continue; used[root] = true; if(rank[root] == 1)small++; else big++; } out.println(Math.abs(big-small)); } public static void main(String[] args) { out.flush(); new Main().solve(); out.close(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } }
RUPC2016 Day2 B SEARIGHT LIVE FES
考察
外側の角の部分から上,横,下に分けて考える。
外側の人がa番目に存在する
内側の人がb番目に存在する
以上を前提とすると
- 外側の人が上(W-1番目まで)に存在する場合,a == bで判定
- 外側の人が横(W+1番目からW+H-2番目まで)に存在する場合,a-2 == bで判定
- 外側の人が下(W+H-2+1番目からW*2+H-2番目まで)に存在する場合,a-4 == bで判定
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.NoSuchElementException; public class Main{ int W,H,N,a,b; public void solve() { W = nextInt(); H = nextInt(); N = nextInt(); int ans = 0; a = 1; b = 1; for(int i = 0;i < N;i++){ int p = nextInt(); if(p==0)a++; else b++; if(a <= W-1){ if(a == b)ans++; }else if(a == W){ continue; }else if(W + 1 <= a && a <= W + H - 2){ if(a - 2 == b)ans++; }else if(a == W + H - 2 + 1){ continue; }else if(a >= W + H){ if(a - 4 == b)ans++; } } out.println(ans); } public static void main(String[] args) { out.flush(); new Main().solve(); out.close(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } }
RUPC2016 Day2 A Gossip
考察
が答え。
つまり最初に情報を持っているアイドルの間隔を2で割った値の最大値が解だが,アイドル1とアイドルNがそれぞれ情報持っているアイドルの中で一番間隔が近いアイドルとの間隔も先程の解の候補に含める。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.NoSuchElementException; public class Main{ int N,M; int[] a; public void solve() { N = nextInt(); M = nextInt(); a = new int[M]; for(int i = 0;i < M;i++)a[i] = nextInt(); int ans = 0; for(int i = 0;i <= M;i++){ if(i == 0)ans = Math.max(ans, a[i]-1); else if(i==M)ans = Math.max(ans,N-a[i-1]); else ans = Math.max(ans, (a[i]-a[i-1])/2); } out.println(ans); } public static void main(String[] args) { out.flush(); new Main().solve(); out.close(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } }
RUPC2016 Day1 F リレー / Relay
考察
基本的には公式解説にある通りの考え方で何となく理解しましたが,オイラーツアーを行う実装がイメージできなかったので,mayokoさんの解説記事のコードを参考にさせていただきました。セグメントツリーは蟻本写経しました。
公式解説よりうまく説明できないと思うので,メモ程度に書いていきます。
まず,求めたいのが木の直径を構成する端点 u,v なので,それらの端点をdouble sweepというアルゴリズムを使って求める。
double sweep
- 木から適当に頂点を決めてそこから最も距離が遠い頂点を v
- vから最も距離が遠い頂点を u
- v と u はその木の直径を構成する頂点となる。つまりv と u の距離がその木の直径となる。
その後 u,v を根としてそれぞれからオイラーツアーを行う。
(u,v)から付け替える辺 e を除いて到達できる最も遠い頂点を決め距離を求める。
オイラーツアーでやること
- depth[v] = 根から頂点vまでの距離
- first[v] = 根から深さ優先探索をして最初に頂点vに到達した時点のそれまでの頂点を訪れた回数
- last[v] = 根から深さ優先探索をして最後に頂点vに到達した時点のそれまでの頂点を訪れた回数
- 深さ優先探索をしつつ頂点を訪れるたびに回数をカウントする
- 初めて訪れた頂点でfirst[v]に訪れた回数を記録,depth[v]に根からの距離を記録
- その頂点から出ていき,もう訪れない時はlast[v]に訪れた回数を記録する
公式解説:
立命合宿 2016 - 立命館大学情報理工学部プロジェクト団体プログラミングコンテスト部門 RiPPro
参考にした記事等:
mayokoex.hatenablog.com
あとは公式解説内にあるリンク先とか参考にしました。
ソースコード
恐らく僕の実装があまり良くないらしくてスタックのサイズを指定しないとRuntime Errorになってしまいます。
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.ArrayList; import java.util.Arrays; import java.util.NoSuchElementException; public class Main implements Runnable{ int N,num,size; int[] depth,first,last,segment; ArrayList<Edge>[] graph; private class Edge{ int to,weight; public Edge(int to,int weight){ this.to = to; this.weight = weight; } } public void init(int n){ size = 1; while(size < n)size<<=1; segment = new int[2*size]; for(int i = 0;i < 2 * size - 1;i++)segment[i]=Integer.MIN_VALUE; } public void update(int k,int a){ k += size - 1; segment[k] = a; while(k > 0){ k = (k - 1) / 2; segment[k] = Math.max(segment[k * 2 + 1], segment[k * 2 + 2]); } } public int query(int a,int b,int k,int l,int r){ if(r <= a || b <= l)return Integer.MIN_VALUE; if(a <= l && r <= b)return segment[k]; int vl = query(a,b,k * 2 + 1,l,(l + r)/2); int vr = query(a,b,k * 2 + 2,(l + r)/2,r); return Math.max(vl, vr); } public int[] dfs(int prev,int now){ int[] ret = {now,0}; for(Edge edge : graph[now]){ if(edge.to != prev){ int[] tmp = dfs(now,edge.to); tmp[1]+=edge.weight; if(ret[1] < tmp[1])ret = tmp; } } return ret; } //木の直径の端点u,vを求める public int[] doubleSweep(){ int u = dfs(-1,0)[0]; int v = dfs(-1,u)[0]; return new int[]{u,v}; } public void eulerTour(int prev,int now,int dis){ depth[now] = dis; first[now] = num++; for(Edge edge : graph[now]){ if(edge.to != prev){ eulerTour(now,edge.to,dis+edge.weight); num++; } } last[now] = num; } @SuppressWarnings("unchecked") public void solve() { N = nextInt(); graph = new ArrayList[N]; for(int i = 0;i < N;i++)graph[i] = new ArrayList<Edge>(); for(int i = 0;i < N-1;i++){ int p = nextInt(); int w = nextInt(); graph[i+1].add(new Edge(p,w)); graph[p].add(new Edge(i+1,w)); } int[] tmp = doubleSweep(); int u = tmp[0]; int v = tmp[1]; int ans = 0; int M = 2 * N; depth = new int[M]; first = new int[M]; last = new int[M]; //u num = 0; eulerTour(-1,u,0); init(2*M); for(int i = 0;i < N;i++){ update(first[i],depth[i]); } for(int i = 0;i < N;i++){ for(Edge edge : graph[i]){ int to = edge.to; if(depth[edge.to] < depth[i])to = i; int vl = query(0,first[to],0,0,size); int vr = query(last[to],M-1,0,0,size); ans = Math.max(ans,Math.max(vl, vr)+edge.weight); } } Arrays.fill(depth, 0); Arrays.fill(first,0); Arrays.fill(last, 0); Arrays.fill(segment, Integer.MIN_VALUE); num = 0; //v eulerTour(-1,v,0); for(int i = 0;i < N;i++){ update(first[i],depth[i]); } for(int i = 0;i < N;i++){ for(Edge edge : graph[i]){ int to = edge.to; if(depth[edge.to] < depth[i])to = i; int vl = query(0,first[to],0,0,size); int vr = query(last[to],M-1,0,0,size); ans = Math.max(ans,Math.max(vl, vr)+edge.weight); } } out.println(ans); } public static void main(String[] args) { new Thread(null,new Main(),"",32*1024 * 1024).start(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } @Override public void run() { out.flush(); new Main().solve(); out.close(); } }
RUPC2016 Day1 E 28
考察
良い整数は各桁に,2または8しか現れないのでまでの正整数はそこまで多くない。
なので、考えられる良い整数はすべて生成しておき、を割り切ることが出来る良い整数だけを抜き出しておく。
抜き出した良い整数の積からNが構成できるかどうかをメモ化全探索する。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.ArrayDeque; import java.util.ArrayList; import java.util.Collections; import java.util.HashMap; import java.util.NoSuchElementException; public class Main { long N; ArrayList<Long> all,sub; HashMap<Long,Integer>[] map; public void createGoodNumber(){ all = new ArrayList<Long>(); ArrayDeque<Long> q = new ArrayDeque<Long>(); q.add(2L); q.add(8L); while(q.size() > 0){ long L = q.poll(); all.add(L); if(L >= (long)1e17)continue; q.add(L * 10 + 2); q.add(L * 10 + 8); } Collections.sort(all); } public int dfs(int i,long n){ if(n == 1){ return 0; } if(map[i].containsKey(n)){ return map[i].get(n); } int ret = -1; for(int j = i;j < sub.size();j++){ if(n % sub.get(j) == 0){ int tmp = dfs(j,n / sub.get(j)); if(tmp+1 > 0){ map[i].put(n, tmp+1); return tmp+1; } } } map[i].put(n,ret); return ret; } @SuppressWarnings("unchecked") public void solve() { N = nextLong(); createGoodNumber(); sub = new ArrayList<Long>(); for(int i = 0;i < all.size();i++){ if(N % all.get(i) == 0)sub.add(all.get(i)); } if(N == 1){ out.println(-1); return; } map = new HashMap[sub.size()]; for(int i = 0;i < sub.size();i++){ map[i] = new HashMap<Long,Integer>(); } if(sub.size() == 0){ out.println(-1); }else{ out.println(dfs(0,N)); } } public static void main(String[] args) { out.flush(); new Main().solve(); out.close(); } /* Input */ private static final InputStream in = System.in; private static final PrintWriter out = new PrintWriter(System.out); private final byte[] buffer = new byte[2048]; private int p = 0; private int buflen = 0; private boolean hasNextByte() { if (p < buflen) return true; p = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) return false; return true; } public boolean hasNext() { while (hasNextByte() && !isPrint(buffer[p])) { p++; } return hasNextByte(); } private boolean isPrint(int ch) { if (ch >= '!' && ch <= '~') return true; return false; } private int nextByte() { if (!hasNextByte()) return -1; return buffer[p++]; } public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = -1; while (isPrint((b = nextByte()))) { sb.appendCodePoint(b); } return sb.toString(); } public int nextInt() { return Integer.parseInt(next()); } public long nextLong() { return Long.parseLong(next()); } public double nextDouble() { return Double.parseDouble(next()); } }