亲宝软件园·资讯

展开

Java 贪心和枚举

撩得Android一次心动 人气:0

笔试技巧:学会根据数据范围猜知识点         

一般1s 时间限制的题目,时间复杂度能跑到 1e8 左右( python 和 java 会少一些,所以建议大家使用c/c++ 做笔试题)。

n 范围 20 以内:
O(n*2^n)
状压搜索 /dfs 暴搜
n 范围 200 以内:
O(n^3)
三维 dp
n 范围 3000 以内:
O(n^2)
二维 dp 背包 枚举 二维前缀和等
n 范围 1e5 以内:
O(n√n)
暴力求因子等
n 范围 1e6 以内:
O(nlogn)
二分答案 使用各种 stl 并查集 欧拉筛等
n 范围 1e7 以内:
O(n)
双指针 线性 dp 差 分 / 前缀和
n 范围 1e14 以内:
O(√n)
求约数和 约数个数

贪心:

贪心指每一步都做出当前最优的选择。一般解决的问题有如下特点:局部最优能导致全局最优。

请注意,贪心并不是万能的!

有n个物品。每个物品有价值v[i],以及重量w[i]。

现在取总重量不超过m的物品,问取的物品价值最大是多少?(背包问题)

枚举:

1.朴素枚举

顾名思义,用for循环枚举所有情况。

2.状压枚举   

借助n进制的性质进行枚举。

适合场景:共有n个物品,每个物品有m种状态,枚举所有物品的所有状态,复杂度为O(m^n)。

二进制状压枚举的写法:

典型场景: 总共有n个数:a1、a2……an,每个数可以取也可以不取,枚举所有方案。      

for(int i=0;i<1<<n;i++){  //i为1到2^n的状态压缩值   2^n
	int p=i;          //先将i取出
	int sum=0;        //用一个变量维护取出的数之和
	for(j=0;j<n;j++){   //转为二进制进行操作
		sum+=p%2*a[j];    //这句话是求取出来的数之和。p%2为对应二进制位
                  		 //这里也可以进行其他操作,不一一举例。
		p/=2;            //求下一个二进制位
     	}
     //这里可以添加想要的操作。
   }

算法题1:

chika和蜜柑(PriorityQueue+Comparator的使用)

难度 ⭐⭐

chika很喜欢吃蜜柑。每个蜜柑有一定的酸度和甜度,chika喜欢吃甜的,但不喜欢吃酸的。

一共有n个蜜柑,chika吃k个蜜柑,将获得所吃的甜度之和与酸度之和。chika想获得尽可能大的甜度总和。如果有多种方案,她希望总酸度尽可能小。

她想知道,最终的总酸度和总甜度是多少?

题目信息:先按甜度降序排序,后按酸度升序排序(保持之前的甜度降序排序优先,酸度升序排序次之) 

输入描述:

第一行有两个正整数n和k,分别代表蜜柑总数和chika吃的蜜柑数量。(1≤k≤n≤200000)

第二行有n个正整数ai,分别代表每个蜜柑的酸度。(1≤ai≤1e9)

第二行有n个正整数bi,分别代表每个蜜柑的甜度。(1≤bi≤1e9)

输出描述:

两个正整数,用空格隔开。分别表示总酸度和总甜度。

示例

输入

3 2

1 3 4

2 2 5

输出

5 7

说明

选择1号和3号蜜柑,总酸度为5,总甜度为7,为最优解。

import java.io.*;
import java.util.*;
public class Main{
    public static class Orange{
            int acid;
            int sweet;
            public Orange(int acid, int sweet){
                this.acid = acid;
                this.sweet = sweet;
            }
            public Orange(){}
        }
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] tmp = br.readLine().split(" ");
        int n = Integer.parseInt(tmp[0]);
        int k = Integer.parseInt((tmp[1]));
        String[] ai = br.readLine().split(" ");
        String[] bi = br.readLine().split(" ");
        //定义一个优先队列,并根据指定的比较器对其元素进行排序。
        PriorityQueue<Orange> queue = new PriorityQueue<>((Orange o1, Orange o2)->{
            //匿名内部类以lambda的形式定义排序规则
            if(o1.sweet == o2.sweet){
                return o1.acid - o2.acid;
            }else{
                return o2.sweet - o1.sweet;
            }
        });
        for(int i = 0; i < n; i++){
            Orange orange = new Orange();
            orange.acid = Integer.parseInt(ai[i]);
            orange.sweet = Integer.parseInt(bi[i]);
            queue.add(orange);
        }
        long totalAcid = 0;
        long totalSweet = 0;
        for(int i = 0; i < k; i++){
            Orange o = queue.poll();
            totalAcid += o.acid;
            totalSweet += o.sweet;
        }
        System.out.println(totalAcid + " " + totalSweet);
    }
}

目的:

了解什么是贪心,当设计到优先级时可以考虑使用PriorityQueue+Comparator。

算法题2:

you和帆船 

难度 ⭐⭐

you的父亲是一名船长。因此you从小就很喜欢大海。这天,她乘着一艘帆船出发了。

大海上有很多宝藏,每个宝藏的坐标是已知的。you的初始坐标是(0,0)。她想探索两个宝藏,然后回到初始点。

you希望航线尽可能短。她想知道,最短航线的长度是多少?

题目信息:两个for循环枚举一下,再保存最短路径即可。

输入描述:

第一行一个正整数n,代表宝藏的数量。(2≤n≤2000)

接下来的n行,每行2个正整数xi,yi,代表第i个宝藏的坐标(-3000≤xi,yi≤3000)

不保证不存在两个宝藏位置相同。意思是,you可以在同一个位置探索这两个宝藏。

输出描述:

最短路线的长度。小数点后保留6位。

示例

输入

2

1 0

0 1

输出

3.414214

说明

import java.io.*;
import java.util.*;
 
class Point{
    double x;
    double y;
}
 
public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        Point[] points = new Point[n];
        for(int i=0;i<n;i++){
            String[] strings = br.readLine().split(" ");
            Point point = new Point();
            point.x = Integer.parseInt(strings[0]);
            point.y = Integer.parseInt(strings[1]);
            points[i] = point;
        }
        double min = Double.MAX_VALUE;//定义最大值,寻找较小值
        //双层遍历枚举
        for (int i=0;i<n-1;i++) {
            for (int j=i+1;j<n;j++) {
                double temp = Math.sqrt(points[i].x*points[i].x + points[i].y*points[i].y) 
                        + Math.sqrt(points[j].x*points[j].x + points[j].y*points[j].y) 
                        + Math.sqrt((points[i].x-points[j].x)*(points[i].x-points[j].x) + (points[i].y-points[j].y)*(points[i].y-points[j].y));
                min = Math.min(min, temp);
            }
        }
        System.out.println(String.format("%.6f", min));
    }
}

目的:

了解什么是枚举,虽然是一个一个列举,但是结合实际还是有优化的方式。

比如此题两层循环只枚举了一半的情况:因为所求的是距离,跟两个端点无关。

思考:

假如不止有两个宝箱需要被获取,那么应该怎么办???下一题可以参考一下。

算法题3: 

数位染色   

难度 ⭐⭐⭐

小红拿到了一个正整数 X 。她可以将其中一些数位染成红色。然后她想让所有染红的数位数字之和等于没染色的数位数字之和。

她不知道能不能达成目标。你能告诉她吗?

输入描述:

一个正整数X ,1<= X <=10^18

输出描述:

如果小红能按要求完成染色,输出"Yes"。否则输出"No"。

示例1

输入

1234567

输出

Yes

说明

将3、4、7染成红色即可,这样3+4+7=1+2+5+6

示例2

输入

23

输出

No

说明

显然无论如何都不能完成染色。

import java.util.*;
import java.io.*;
public class Main{
    public static void main(String[] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Long x= Long.parseLong(br.readLine());
        //循环0到2^18来展现所有的可能性
        for(int i=0;i<1<<19;i++){
            long p=i,s1=0,s2=0,temp=x;
 
            //将p拟化为2进制,通过j来寻尾。寻一次p就对应的二进制数就少一位。
            for(int j=0;j<19;j++){
                if(p%2==1)s1+=temp%10;
                else s2+=temp%10;
                p/=2;
                temp/=10;
            }
            if(s1==s2){
                System.out.println("Yes");
                System.exit(0);
            }
        }
        System.out.println("No");
    }
}

这题使用的是状压枚举

只有两种状态就拟成2进制,假如有3种状态就拟成3进制(上面的代码会有些许改变,n种状态也一样)

 for(int i=0;i<1<<19;i++)  
//修改成
 for(int i=0;i<19;i++) p1[i]=p1[i-1]*3;
 for(int i=0;i<p1[i];i++){}

当然这题也可以使用背包或dfs来解答。

算法题4: 

ranko的手表  

难度 ⭐⭐⭐⭐

ranko 的手表坏了,正常应该显示 xx:xx 的形式(4 个数字),比如下午 1 点半应该显示 13:30 ,但现在经常会有一些数字有概率无法显示。

ranko 在 t1 时刻看了下时间,过了一段时间在 t2 时刻看了下时间。她想知道, t1  和t2这两个时刻之间相距的时间的最大值和最小值是多少?

保证t1在t2之前(且t1和t2不等)。 t1和t2在同一天的 00:00 到 23:59 之间。 

输入描述:

两行输入两个时间,为 xx:xx 的形式。其中 x 为数字或者字符 '?' ,问号代表这个数字没有显示。保证输入是合法的。

输出描述:

一行输出两个整数,分别代表 t1 和 t2 相距时间的最小值和最大值(单位分钟)。

示例

输入

18:0?

2?:1?

输出

121 319

说明

相距最小的时间为 18:09 到 20:10 ,相距121分钟。

相距最大的时间为 18:00 到 23:19 ,相距319分钟。

import java.util.*;
import java.io.*;
public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s1 = br.readLine();
        String s2 = br.readLine();
        ArrayList<Integer> a1 = new ArrayList<>();
        ArrayList<Integer> a2 = new ArrayList<>();
        for(int i = 0; i < 60*24; i++){
            int hour = i/60, minute = i%60;
            if((hour/10 == s1.charAt(0)-'0' || s1.charAt(0) == '?') &&
               (hour%10 == s1.charAt(1)-'0' || s1.charAt(1) == '?') &&
               (minute/10 == s1.charAt(3)-'0' || s1.charAt(3) == '?') &&
               (minute%10 == s1.charAt(4)-'0' || s1.charAt(4) == '?')) a1.add(i);
            if((hour/10 == s2.charAt(0)-'0' || s2.charAt(0) == '?') &&
               (hour%10 == s2.charAt(1)-'0' || s2.charAt(1) == '?') &&
               (minute/10 == s2.charAt(3)-'0' || s2.charAt(3) == '?') &&
               (minute%10 == s2.charAt(4)-'0' || s2.charAt(4) == '?'))a2.add(i);
        }
        int min= Integer.MAX_VALUE, max = Integer.MIN_VALUE;
        for(int i = 0; i<a1.size();i++){
            for(int j = 0; j<a2.size();j++){
                if(a1.get(i)<a2.get(j)){
                    min = Math.min(min,a2.get(j)-a1.get(i));
                    max = Math.max(max,a2.get(j)-a1.get(i));
                }
            }
        }
        System.out.print(min + " " + max);
    }
}

假如此题不使用枚举,则会限定很多条件。还不如直接都列举出来

加载全部内容

相关教程
猜你喜欢
用户评论