[leetcode] dp

1-D DP:

Climbing Stairs:

Naive 就是sum[0]; if (n==0) sum[0]++;

然后climbstairs(n-1)

climbstairs(n-2)


    public int climbStairs(int n) {
        if (n<=0) return 0;
        if (n==1) return 1;
        int[] d = new int[n+1];
        d[0]=1;
        d[1]=1;
        for (int i=2;i<=n;i++) {
            d[i]=d[i-1]+d[i-2];
        }
        return d[n];
    }


Decode Ways:

分类写清楚

char在string里的pos和在d[]里的位置搞清楚(differ by 1)

    public int numDecodings(String s) {
        if (s.length()==0) return 0;
        if (s.charAt(0)=='0') return 0;
        if (s.length()==1) return 1;
        int num0=1;
        int num1=1;
        int num2=1;
        for (int i=2;i<=s.length();i++) {
            char a = s.charAt(i-1);
            char b = s.charAt(i-2);
            if (a=='0') {
                if (b=='1' ||b=='2') num2=num0;
                else return 0;
            } else if (a>='1' && a<='6') {
                if (b=='1' ||b=='2') num2=num1+num0;
                else num2=num1;
            } else {
                if (b=='1') num2=num1+num0;
                else num2=num1;
            }
            num0=num1;
            num1=num2;
        }
        return num2;
    }

Unique Binary Search Trees

catalan number: d[i]=sum(d[j]*d[i-j-1]) j=0..i-1

    public int numTrees(int n) {
        if (n==0) return 0;
        if (n==1) return 1;
        int[] d = new int[n+1];
        d[0]=1;
        for (int i=1;i<=n;i++) {
            for (int j=0;j<i;j++) {
                d[i]+=d[j]*d[i-j-1];
            }
        }
        return d[n];
    }


Maximum Subarray

    public int maxSubArray(int[] A) {
        if (A==null||A.length==0) return 0;
        if (A.length==1) return A[0];
        int max = A[0];
        int local = A[0];
        for (int i=1;i<A.length;i++) {
            if (local>0) local = local+A[i];
            else local = A[i];
            max = Math.max(local, max);
        }
        return max;
    }



Maximum Product Subarray

* localMin会被改掉,所以要用temp啊~

    public int maxProduct(int[] A) {
        if (A==null||A.length==0) return 0;
        int max = A[0];
        int localMin = A[0];
        int localMax = A[0];
        for (int i=1;i<A.length;i++) {
            int temp = localMin;
            localMin = Math.min(A[i]*localMax, Math.min(A[i],A[i]*localMin));
            localMax = Math.max(A[i]*temp, Math.max(A[i],A[i]*localMax));
            max = Math.max(localMax,max);
        }
        return max;
    }


Best Time to Buy and Sell Stock、

    public int maxProfit(int[] prices) {
        if (prices==null||prices.length==0) return 0;
        int local = 0;
        int max = 0;
        for (int i=0;i<prices.length-1;i++) {
            local = Math.max(local-prices[i]+prices[i+1],0);
            max = Math.max(local,max);
        }
        return max;
    }

Jump Game 

* i<=max

    public boolean canJump(int[] A) {
        if (A==null || A.length==0 ||A.length==1) return true;
        if (A[0]==0) return false;
        int max = A[0];
        for (int i=1;i<A.length && i<=max;i++) {
            max = max>A[i]+i ? max : A[i]+i;
        }
        if (max>=A.length-1) return true;
        return false;
    }


Jump Game II (find min step)

    public int jump(int[] A) {
        if (A==null|| A.length==0) return 0;
        int lastReach = 0;
        int reach = 0;
        int step = 0;
        for (int i=0;i<=reach&&i<A.length;i++) {
            if (i>lastReach) {
                step++;
                lastReach=reach;
            }
            reach = Math.max(reach,A[i]+i);
        }
        if (reach<A.length-1) return 0;
        return step;

Longest Valid Parentheses 
从后往前走 d[i]代表以i开头的最长valid parentheses
递推式: 如果charAt(i)='(' && charAt(end)=')' 就比后面的加2
*如果end后面也有valid parentheses 就需要加上 (ie. (())()  )

    public int longestValidParentheses(String s) {
        if (s==null||s.length()<=1) return 0;
        int max = 0;
        int[] dp = new int[s.length()];
        for (int i=s.length()-2;i>=0;i--) {
            int end = dp[i+1]+i+1;
            if (end<s.length() && s.charAt(i)=='(' && s.charAt(end)==')') {
                dp[i]=dp[i+1]+2;
                if (end+1<s.length()) dp[i]+=dp[end+1];
            }
            max=Math.max(max, dp[i]);
        }
        return max;
    }


Triangle
1. 从上往下走:sum[i]=min(sum[i],sum[i-1])+n[i] 需要单独处理头和尾,最后还要走一遍找max
2. 从下往上走:sum[i]=min(sum[i],sum[i+1])+n[i] 不用处理头尾 代码简洁一些
    public int minimumTotal(List<List<Integer>> triangle) {
        if (triangle==null||triangle.size()==0) return 0;
        int[] sum = new int[triangle.get(triangle.size()-1).size()];
        for (int i=triangle.size()-1;i>=0;i--) {
            List<Integer> layer = triangle.get(i);
            for (int j=0;j<=i;j++) {
                if (i==triangle.size()-1) sum[j]=layer.get(j);
                else sum[j]=Math.min(sum[j],sum[j+1])+layer.get(j);
            }
        }
        return sum[0];
    }


2-D DP:

Palindrome Partitioning II 

1. 查palindrome的dp: 一个boolean[i][j]代表从i到j是不是palindrome。 填满右上半部分的matrix,diagonal为1。复杂度n^2

            if (s.charAt(i)==s.charAt(j) && (j-i<2 || res[i+1][j-1])) res[i][j]=true;


2. 查min cut的dp: 一个int[]代表到i的min cut。复杂度n^2 (注意j=0的边界情况)

                if (dict[j][i]) d[i]=Math.min(d[i], (j==0 ? 0 : d[j-1])+1);

    public int minCut(String s) {
        if(s==null || s.length()==0) return 0;
        boolean[][] dict = getDict(s);
        int[] d = new int[s.length()];
        for (int i=0; i<s.length(); i++) {
            d[i]=i+1;
            for (int j=0;j<=i;j++) {
                if (dict[j][i]) d[i]=Math.min(d[i], (j==0 ? 0 : d[j-1])+1);
            }
        }
        return d[s.length()-1]-1;
    }
    
    private boolean[][] getDict(String s) {
        boolean[][] res = new boolean[s.length()][s.length()];
        for (int i=s.length()-1;i>=0; i--) {
            for (int j=i; j<s.length(); j++) 
            if (s.charAt(i)==s.charAt(j) && (j-i<2 || res[i+1][j-1])) res[i][j]=true;
        }
        return res;
    }

Edit Distance

递推式:d[i][j]=min(d[i][j-1],d[i-1][j],d[i-1][j-1])+1 <==== word1的i==word2的j
d[i][j]=d[i-1][j-1] <=== word1的i!=word2的j

    public int minDistance(String word1, String word2) {
        if (word1==null||word2==null) return 0;
        if (word1.length()==0||word2.length()==0) 
        return word1.length()==0 ? word2.length():word1.length();
        int[][] d = new int[word1.length()][word2.length()];
        boolean flag1=false;
        boolean flag2=false;
        for (int i=0;i<word2.length();i++) {
            if (word1.charAt(0)==word2.charAt(i) && !flag1) {
                flag1 = true;
                d[0][i]=i==0 ? 0 : d[0][i-1];
            } else d[0][i]= i==0 ? 1 : d[0][i-1]+1;
        }
        for (int i=0;i<word1.length();i++) {
            if (word2.charAt(0)==word1.charAt(i) && !flag2) {
                flag2=true;
                d[i][0]= i==0 ? 0 : d[i-1][0];
            } else d[i][0] = i==0 ? 1 : d[i-1][0]+1;
        }
        for (int i=1;i<word1.length();i++) {
            for (int j=1;j<word2.length();j++) {
                if (word1.charAt(i)==word2.charAt(j)) 
                d[i][j]=d[i-1][j-1];
                else d[i][j] = Math.min(d[i][j-1],Math.min(d[i-1][j],d[i-1][j-1]))+1;
            }
        }
        return d[word1.length()-1][word2.length()-1];
    }

Minimum Path Sum (mxn grid 从 左上到右下 的最小路径)

//h[i][j]=min(h[i+1][j],h[i][j+1])+grid[i][j]

    public int minPathSum(int[][] grid) {
        if (grid==null||grid.length==0||grid[0].length==0) return 0;
        int row = grid.length;
        int col = grid[0].length;
        int[][] h = new int[grid.length][grid[0].length];
        for (int i=row-1;i>=0;i--) 
        h[i][col-1]= i==grid.length-1? grid[i][col-1] : h[i+1][col-1]+grid[i][col-1];
        for (int i=col-2;i>=0;i--) 
        h[row-1][i]= h[row-1][i+1]+grid[row-1][i];
        for (int i=row-2;i>=0; i--) {
            for (int j=col-2;j>=0;j--) {
                h[i][j]=Math.min(h[i+1][j], h[i][j+1])+grid[i][j];
            }
        }
        return h[0][0];
    }

Word Break

//a:i-0
//if h[a] true && substring(a+1, i+1) in dict ==> h[i] true

注意第一个case,两个解决办法:

1. h[len+1], h[0]=true,代表前i个字母能不能word break

2. j=-1;j<i;j++, 然后加if j==-1?true:h[j]

    public boolean wordBreak(String s, Set<String> dict) {
        if (s==null||dict==null||dict.isEmpty()) return false;
        boolean[] h = new boolean[s.length()+1];
        h[0]=true;
        for (int i=1;i<=s.length();i++) {
            for (int j=0;j<i;j++) {
                if (h[j] && dict.contains(s.substring(j,i))) {
                    h[i]=true;
                }
            }
        }
        return h[s.length()];
    }

Word Break II (输出所有可能的solution)

*复杂度取决于结果数量,直接brute force对于结果多的情况跟dp差不多

brute force:

    public List<String> wordBreak(String s, Set<String> dict) {
        List<String> res = new ArrayList<String>();
        if (s==null||dict==null||dict.isEmpty()) return res;
        helper(s, 0, new ArrayList<String>(), dict,res);
        return res;
    }
    
    private void helper(String s, int start, List<String> item, Set<String> dict, List<String> res) {
        if (start==s.length()) {
            res.add(convert(item));
            return;
        }
        for (int i=start;i<s.length();i++) {
            String sub = s.substring(start,i+1);
            if (dict.contains(sub)) {
                item.add(sub);
                helper(s, i+1, item, dict, res);
                item.remove(item.size()-1);
            }
        }
    }
    
    private String convert(List<String> item) {
        StringBuilder sb = new StringBuilder();
        for (String s: item) {
            sb.append(s);
            sb.append(" ");
        }
        sb.deleteCharAt(sb.length()-1);
        return sb.toString();
    }

dp + map

先做dp,做的路上把最后的word放到map对应的i里;如果最后可以达到,就做个dfs把所有结果都添加了

注意:dfs时的空格要做check,item是空的时不要加空格

    public List<String> wordBreak(String s, Set<String> dict) {
        List<String> res = new ArrayList<String>();
        Map<Integer, List<String>> map = new HashMap<Integer, List<String>>();
        if (s==null||dict==null||dict.isEmpty()) return res;
        boolean[] h = new boolean[s.length()];
        for (int i=0;i<s.length();i++) {
            for (int j=-1;j<i;j++) {
                boolean pre = j==-1? true:h[j];
                if (pre && dict.contains(s.substring(j+1,i+1))) {
                    h[i]=true;
                    if (map.containsKey(i)) map.get(i).add(s.substring(j+1,i+1));
                    else {
                        List<String> l = new ArrayList<String>();
                        l.add(s.substring(j+1,i+1));
                        map.put(i, l);
                    }
                }
            }
        }
        if (!h[h.length-1]) return res;
        dfs(map,h,h.length-1, "", res);
        return res;
    }
    
    private void dfs(Map<Integer,List<String>> map, boolean[] h, int end, String item, List<String> res) {
        if (end<0) {
            res.add(item);
            return;
        }
        List<String> list = map.get(end);
        for (String word: list) {
            dfs(map,h,end-word.length(),word+ (item.length()==0? "": " ")+item,res);
        }
    }

distinct subsequences

dp[i][j]=S[i]==T[j]? dp[i-1][j-1] + dp[i-1][j] : dp[i-1][j]  (ie. rabb 和 rab, 如果s[3]==t[2], 数量就是没有s[3]那个b的结果([i-1][j])+ 加上s[3]和t[2]的结果([i-1][j-1])

    R A B R <==T

R 1  0  0  0

A 1  1  0  0 

B 1  1  1  0

B 1  1  2  0

^

S

    public int numDistinct(String S, String T) {
        if (S.length()==0 || T.length()==0) return 0;
        int[][] dp = new int[S.length()][T.length()];
        char[] A = S.toCharArray();
        char[] B = T.toCharArray();
        int count = 0;
        for (int i=0;i<A.length;i++) dp[i][0]=A[i]==B[0]?++count:count;
        for (int i=1;i<A.length;i++) {
            for (int j=1;j<B.length;j++) dp[i][j]=A[i]==B[j]?dp[i-1][j-1]+dp[i-1][j]:dp[i-1][j];
        }
        return dp[S.length()-1][T.length()-1];
    }


maximal rectangle (o(m*n*n)

把每一行一刀切了,当作矩阵的底面,然后保存它上面最多有几个连着的1,这样跑一遍整个矩阵得到res[][]

然后对矩阵里的每个位置进行左右的expand,知道宽度变窄(就是以此为最高以此为底的矩阵),update最大值 (这一步要o(m*n*n))

(其实还有优化 就是用stack来做每一层的面积(largest histogram area)会是 m*n)

    public int maximalRectangle(char[][] rec) {
    	int max = 0;
    	if (rec.length==0 || rec[0].length==0) return 0;
    	int[][] res = new int[rec.length][rec[0].length];
    	for (int j=0;j<rec[0].length;j++) {
    		res[0][j]=(rec[0][j] - '0');
    	}
    	
    	for (int i=1;i<rec.length;i++) {
    		for (int j=0;j<rec[0].length;j++) {
    			res[i][j]=rec[i][j]=='1'? (res[i-1][j]+1) : 0;
    		}
    	}
    	
    	for (int i=0;i<rec.length;i++) {
    		for (int j=0;j<rec[0].length;j++) {
    			if (res[i][j]!=0) {
    				int l = j;
    				int r = j;
    				while (l>0 || r<rec[0].length-1) {
    					boolean done = true;
    					if (l>0 && res[i][l-1]>=res[i][j]) {
    						l--;
    						done = false;
    					}
    					if (r<rec[0].length-1 && res[i][r+1]>=res[i][j]) {
    						r++;
    						done = false;
    					}
    					if (done) break;
    				}
    				int min = r-l+1;
    				max = Math.max(max, min*res[i][j]);
    			}
    		}
    	}
    	return max;
    }


largest rectangle in histogram (o(n^2))

1. 取每个i作为最高高度来expand (while loop, 然后记得skip掉height[i]==height[i+1]的

2. 穷举两个边界,但不是完全穷举,这里可以省掉一部分右右边界。对于i,如果A[i] <= A[i + 1],即当前递增,则无论对于哪个左边界,选i+1作为右边界都比选i作为右边界合适(宽本来就多一个,然后高还更高,所以肯定选右边的面积大)。所以就算i+1作为右边界的所有左边界配对就行,没必要算i作为右边界的了。所以递推下去,只有A[i] > A[i + 1]的时候,在递减,说明拿A[i]和拿A[i + 1]不一定谁大了,这时候算一下A[i]作为右边界的左边界所有可能,就是一个local peak。这题还有一个O(n)的解法,但是暂时懒的看了。下次再看。

public int largestRectangleArea(int[] height) {
    int res = 0;
    for (int i = height.length - 1; i >= 0; i--) {
        if (i == height.length - 1 || height[i] > height[i + 1]) { //A[i] is a local peak
            int min = Integer.MAX_VALUE;
            for (int j = i; j >= 0; j--) {
                min = Math.min(min, height[j]);
                res = Math.max(res, (i - j + 1) * min);
            }
        }
    }
    return res;
}


candy

左边扫一遍保证他们拿的比左边的多

右边扫一遍保证他们拿的比右边的多


    public int candy(int[] ratings) {
        if (ratings.length==0) return 0;
        int[] dp = new int[ratings.length];
        int res = 0;
        dp[0]=1;
        for (int i=1; i<dp.length;i++) {
            if (ratings[i]>ratings[i-1]) dp[i]=dp[i-1]+1;
            else dp[i]=1;
        }
        int[] dp2 = new int[ratings.length];
        dp2[dp2.length-1] = 1;
        for (int i=dp2.length-1;i>=0;i--) {
            if (i!=dp2.length-1 && ratings[i]>ratings[i+1]) {
                dp2[i]=dp2[i+1]+1;
            } else dp2[i]=1;
            res += Math.max(dp[i],dp2[i]);
        }
        return res;
    }




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值