OJ-2:区间问题【九度1554】
- 题目描述：
-
给定一个数组，判断数组内是否存在一个连续区间，使其和恰好等于给定整数k。
- 输入：
-
输入包含多组测试用例，每组测试用例由一个整数n(1<=n<=10000)开头，代表数组的大小。
接下去一行为n个整数，描述这个数组，整数绝对值不大于100。
最后一行为一个整数k(大小在int范围内)。
- 输出：
-
对于每组测试用例，若存在这个连续区间，输出其开始和结束的位置，s，e(s <= e)。
若存在多个符合条件的输出，则输出s较小的那个，若仍然存在多个，输出e较小的那个。
若不存在，直接输出"No"。
- 样例输入：
-
5 -1 2 3 -4 9 5 3 -1 2 -3 7 2 -1 1 0
- 样例输出：
-
2 3 No 1 2
我的解答：
package jobdu.wangdao; import java.util.Scanner; /** * 题目描述： 给定一个数组，判断数组内是否存在一个连续区间，使其和恰好等于给定整数k。 * */ public class Question1554 { private static void solve(int[] array, int k) { boolean find = false; for (int i = 0; i < array.length; i++) { if (!find) { int sum = 0; for (int j = i; j < array.length; j++) { sum += array[j]; if (sum == k) { find = true; System.out.println((i + 1) + " " + (j + 1)); break; } } } } if (!find) System.out.println("No"); } public static void main(String[] args) { @SuppressWarnings("resource") Scanner scanner = new Scanner(System.in); while (scanner.hasNext()) { int n = scanner.nextInt(); int[] array = new int[n]; for (int i = 0; i < n; i++) { array[i] = scanner.nextInt(); } int k = scanner.nextInt(); solve(array, k); } } }
这个解答用的是暴力o(n*n)破解，提交后九道提示超时。
用动态规划的思想，记录从a[0]开始到a[i]的和存入sum[i]，那么当sum[i]-sum[j]=k的时候从a[j+1]到a[i]的和就为k。
package jobdu.wangdao; import java.io.IOException; import java.io.StreamTokenizer; import java.util.ArrayList; import java.util.Collections; import java.util.HashMap; import java.util.List; import java.util.Map; /** * 题目描述： 给定一个数组，判断数组内是否存在一个连续区间，使其和恰好等于给定整数k。 * */ public class Question1554 { /** * 暴力求和比较 * * @param array * @param k * @return */ public static Section solve(int[] array, int k) { boolean find = false; for (int i = 0; i < array.length; i++) { if (!find) { int sum = 0; for (int j = i; j < array.length; j++) { sum += array[j]; if (sum == k) { find = true; System.out.println((i + 1) + " " + (j + 1)); return new Section(i + 1, j + 1); } } } } if (!find) System.out.println("No"); return null; } public static Section solve2(int[] array, int k) { int[] sumList = new int[array.length];// 先计算出前0-i个数的和存入sumList Map<Integer, List<Integer>> map = new HashMap<Integer, List<Integer>>(); // 记录前i个数的和和值相同的下标 int sum = 0; for (int i = 0; i < array.length; i++) { sum = sum + array[i]; sumList[i] = sum; if (map.containsKey(sum)) {// 将和的下标存入map List<Integer> indexList = map.get(sum); indexList.add(i); } else { List<Integer> indexList = new ArrayList<Integer>(); indexList.add(i); map.put(sum, indexList); } } boolean find = false; List<Section> list = new ArrayList<Section>(); for (int j = 0; j < sumList.length; j++) { int sumi = sumList[j];// 当前遍历的前j个数的和为sumi if (sumi == k) {// 如果前j个数的和就是k,那么直接为一组解 Section node = new Section(1, (j + 1)); list.add(node);// 加入解集合 find = true; continue; } int temp = sumi + k;// 计算出对应的和已保证a[i]-a[j]=k if (map.containsKey(temp) && map.get(temp).size() > 0) { // 如果存在a[i]-a[j]=k，那么从第j+1个数到第i个数之和就为k,这里下标从0开始 int start = j + 2; int end = -1; for (Integer integer : map.get(temp)) {// 从和相同的下标中选离j最近比i大的下标作为最后的解 if (integer >= j) { end = integer + 1; break; } } if (end >= 0) { list.add(new Section(start, end)); find = true; } } } if (!find) System.out.println("No"); else { Collections.sort(list);// 对可能的解做一次排序，找到符合题意的解 Section section = list.get(0); System.out.println(section.start + " " + section.end); return section; } return null; } /** * 确保List的排序 * */ public static class Section implements Comparable<Section> { int start, end; public Section(int start, int end) { this.start = start; this.end = end; } @Override public int compareTo(Section o) { if (this.start == o.start) return this.end - o.end; return this.start - o.start; } } public static void main(String[] args) throws IOException { @SuppressWarnings("deprecation") StreamTokenizer st = new StreamTokenizer(System.in); while (st.nextToken() != StreamTokenizer.TT_EOF) { int n = (int) st.nval; int[] array = new int[n]; for (int i = 0; i < n; i++) { st.nextToken(); array[i] = (int) st.nval; } st.nextToken(); int k = (int) st.nval; solve2(array, k); } } }
其中一开始使用Scanner的方法作为输入，优化许久还是超时，查看其它成功代码采用的是StreamTokenizer方式作为输入。也改为StreamTokenizer方式后，提交通过。
