【Codeforces 339C】Xenia and Weights
阿新 • • 發佈:2019-03-10
狀態 put 記憶 打印 動態規劃 i+1 throws iter import
【鏈接】 我是鏈接,點我呀:)
【題意】
在天平上放砝碼
你要在左邊放一下然後到右邊放一下
一直重復這樣放m次
每次你放在其中一邊都要讓另外一邊的重量比你少
你可以用1~10中的某些砝碼
問你要怎樣放才行,或者告知系統不能放m次
【題解】
動態規劃
設dp[i][j][k]表示第i輪結束之後,左邊右邊的重量差的絕對值為j,最後一個放的砝碼重量為k的情況能否達到
枚舉一下每次用哪種砝碼(只要不和之前一個狀態最後一個用的一樣就好)做一下轉移即可。
(倒推然後寫一個記憶化搜索可能更方便,因為可以直接打印出最後的結果
【代碼】
import java.io.*; import java.util.*; public class Main { static InputReader in; static PrintWriter out; public static void main(String[] args) throws IOException{ //InputStream ins = new FileInputStream("E:\\rush.txt"); InputStream ins = System.in; in = new InputReader(ins); out = new PrintWriter(System.out); //code start from here new Task().solve(in, out); out.close(); } static int N = 10; static int M = 1000; static class Task{ String s; int a[] = new int[N+10],n; boolean can[][][] = new boolean[M+10][N+10][N+10]; int pre[][][] = new int[M+10][N+10][N+10]; int m; void dfs(int dep,int delta,int last) { if (dep==0) return; int prelast = pre[dep][delta][last]; dfs(dep-1,last-delta,prelast); out.print(last+" "); } public void solve(InputReader in,PrintWriter out) { s = in.next(); m = in.nextInt(); for (int i = 1;i <= 10;i++) if (s.charAt(i-1)=='1') { a[++n] = i; } can[0][0][0] = true; for (int i = 0;i < m;i++) for (int delta = 0;delta <= N;delta++) for (int j = 0;j <= N;j++) if (can[i][delta][j]==true) { for (int j2 = 1;j2 <= n;j2++) if (a[j2]>delta && a[j2]!=j) { can[i+1][a[j2]-delta][a[j2]] = true; pre[i+1][a[j2]-delta][a[j2]] = j; } } for (int delta = 0;delta <= N;delta++) for (int j = 0;j <= N;j++) if (can[m][delta][j]) { out.println("YES"); dfs(m,delta,j); return; } out.println("NO"); } } static class InputReader{ public BufferedReader br; public StringTokenizer tokenizer; public InputReader(InputStream ins) { br = new BufferedReader(new InputStreamReader(ins)); tokenizer = null; } public String next(){ while (tokenizer==null || !tokenizer.hasMoreTokens()) { try { tokenizer = new StringTokenizer(br.readLine()); }catch(IOException e) { throw new RuntimeException(e); } } return tokenizer.nextToken(); } public int nextInt() { return Integer.parseInt(next()); } } }
【Codeforces 339C】Xenia and Weights