csp ccf 201612-3 权限查询java 100分

问题描述

试题编号：	201612-3
试题名称：	权限查询
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　授权 (authorization) 是各类业务系统不可缺少的组成部分，系统用户通过授权机制获得系统中各个模块的操作权限。 　　本题中的授权机制是这样设计的：每位用户具有若干角色，每种角色具有若干权限。例如，用户 david 具有 manager 角色，manager 角色有 crm:2 权限，则用户 david 具有 crm:2 权限，也就是 crm 类权限的第 2 等级的权限。 　　具体地，用户名和角色名称都是由小写字母组成的字符串，长度不超过 32。权限分为分等级权限和不分等级权限两大类。分等级权限由权限类名和权限等级构成，中间用冒号“:”分隔。其中权限类名也是由小写字母组成的字符串，长度不超过 32。权限等级是一位数字，从 0 到 9，数字越大表示权限等级越高。系统规定如果用户具有某类某一等级的权限，那么他也将自动具有该类更低等级的权限。例如在上面的例子中，除 crm:2 外，用户 david 也具有 crm:1 和 crm:0 权限。不分等级权限在描述权限时只有权限类名，没有权限等级（也没有用于分隔的冒号）。 　　给出系统中用户、角色和权限的描述信息，你的程序需要回答多个关于用户和权限的查询。查询可分为以下几类： 　　* 不分等级权限的查询：如果权限本身是不分等级的，则查询时不指定等级，返回是否具有该权限； 　　* 分等级权限的带等级查询：如果权限本身分等级，查询也带等级，则返回是否具有该类的该等级权限； 　　* 分等级权限的不带等级查询：如果权限本身分等级，查询不带等级，则返回具有该类权限的等级；如果不具有该类的任何等级权限，则返回“否”。 输入格式 　　输入第一行是一个正整数  p，表示不同的权限类别的数量。紧接着的  p 行被称为 P 段，每行一个字符串，描述各个权限。对于分等级权限，格式为 <category>:<level>，其中 <category> 是权限类名，<level> 是该类权限的最高等级。对于不分等级权限，字符串只包含权限类名。 　　接下来一行是一个正整数  r，表示不同的角色数量。紧接着的  r 行被称为 R 段，每行描述一种角色，格式为 　　<role> <s> <privilege 1> <privilege 2> ... <privilege s> 　　其中 <role> 是角色名称，<s> 表示该角色具有多少种权限。后面 <s> 个字符串描述该角色具有的权限，格式同 P 段。 　　接下来一行是一个正整数  u，表示用户数量。紧接着的  u 行被称为 U 段，每行描述一个用户，格式为 　　<user> <t> <role 1> <role 2> ... <role t> 　　其中 <user> 是用户名，<t> 表示该用户具有多少种角色。后面 <t> 个字符串描述该用户具有的角色。 　　接下来一行是一个正整数  q，表示权限查询的数量。紧接着的  q 行被称为 Q 段，每行描述一个授权查询，格式为 <user> <privilege>，表示查询用户 <user> 是否具有 <privilege> 权限。如果查询的权限是分等级权限，则查询中的 <privilege> 可指定等级，表示查询该用户是否具有该等级的权限；也可以不指定等级，表示查询该用户具有该权限的等级。对于不分等级权限，只能查询该用户是否具有该权限，查询中不能指定等级。 输出格式 　　输出共  q 行，每行为 false、true，或者一个数字。false 表示相应的用户不具有相应的权限，true 表示相应的用户具有相应的权限。对于分等级权限的不带等级查询，如果具有权限，则结果是一个数字，表示该用户具有该权限的（最高）等级。如果用户不存在，或者查询的权限没有定义，则应该返回 false。 样例输入 3 crm:2 git:3 game 4 hr 1 crm:2 it 3 crm:1 git:1 game dev 2 git:3 game qa 1 git:2 3 alice 1 hr bob 2 it qa charlie 1 dev 9 alice game alice crm:2 alice git:0 bob git bob poweroff charlie game charlie crm charlie git:3 malice game 样例输出 false true false 2 false true false true false 样例说明 　　样例输入描述的场景中，各个用户实际的权限如下： 　　* 用户 alice 具有 crm:2 权限 　　* 用户 bob 具有 crm:1、git:2 和 game 权限 　　* 用户 charlie 具有 git:3 和 game 权限 　　* 用户 malice 未描述，因此不具有任何权限 评测用例规模与约定 　　评测用例规模： 　　* 1 ≤  p,  r,  u ≤ 100 　　* 1 ≤  q ≤ 10 000 　　* 每个用户具有的角色数不超过 10，每种角色具有的权限种类不超过 10 　　约定： 　　* 输入保证合法性，包括： 　　1) 角色对应的权限列表（R 段）中的权限都是之前（P 段）出现过的，权限可以重复出现，如果带等级的权限重复出现，以等级最高的为准 　　2) 用户对应的角色列表（U 段）中的角色都是之前（R 段）出现过的，如果多个角色都具有某一分等级权限，以等级最高的为准 　　3) 查询（Q 段）中的用户名和权限类名不保证在之前（U 段和 P 段）出现过 　　* 前 20% 的评测用例只有一种角色 　　* 前 50% 的评测用例权限都是不分等级的，查询也都不带等级

字符串处理的感觉特别容易出错，尤其是情况考虑不周全！所以，如果你不确定完全能做对，可以先看题目要求处理简单的输入数据，不容易出错！

以下是代码 ，评测100分，可以直接提交！

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
//100分
public class Main{
	public static void main(String[] args) throws Exception {
		Scanner scanner = new Scanner(System.in);
		int p = scanner.nextInt();//权限数量
		scanner.nextLine();
		Map<Integer,Privilege> privileges = new HashMap<>();//保存权限
		for(int i = 0; i < p;i++) {
			String str = scanner.nextLine();
			Privilege privilege = new Privilege();
			if(str.contains(":")) {
				String[] strs = str.split(":");
				privilege.name = strs[0];
				privilege.range = Integer.parseInt(strs[1]);
			} else {
				privilege.name = str;
			}
			privileges.put(privilege.name.hashCode(), privilege);
		}
		//角色处理
		int r = scanner.nextInt();//角色数量
		Map<Integer,Role> roles = new HashMap<>();
		scanner.nextLine();
		for(int i = 0; i < r; i++) {
			// rols s p1 p2 p3...
			String roleStr = scanner.nextLine();
			Role role = new Role();
			String[] strs = roleStr.split(" ");
			role.roleName = strs[0];
			role.privilegeNum = Integer.parseInt(strs[1]);
			for(int j = 2;j < strs.length;j++) {
				RolePrivilege rp = new RolePrivilege();
				if(strs[j].contains(":")) {
					String[] strss = strs[j].split(":");
					RolePrivilege existP = role.privileges.get(strss[0].hashCode());
					//如果已经存在带某个等级的权限那么更新为最大等级的权限就行了
					if(existP!=null) {
						int cc = Integer.parseInt(strss[1]);
						existP.level = cc > existP.level? cc:existP.level;
						continue;
					}
					rp.privilege = privileges.get(strss[0].hashCode());
					rp.level = Integer.parseInt(strss[1]);
				} else {
					rp.privilege = privileges.get(strs[j].hashCode());
					rp.level = -1;
				}
				role.privileges.put(rp.privilege.name.hashCode(),rp);
			}
			roles.put(role.roleName.hashCode(),role);
		}
		Role role = ((Role)(roles.values().toArray()[0]));
		RolePrivilege ppp = (RolePrivilege) role.privileges.values().toArray()[0];
		//用户处理
		int u = scanner.nextInt();
		scanner.nextLine();
		Map<Integer,User> users = new HashMap<>();
		for(int i = 0; i < u;i++) {
			String userStr = scanner.nextLine();
			User user = new User();
			String[] strs = userStr.split(" ");
			user.userName = strs[0];
			user.roleNum = Integer.parseInt(strs[1]);
			for(int j = 2; j < strs.length;j++) {
				Role ro = roles.get(strs[j].hashCode());
				user.roles.put(ro.roleName.hashCode(),ro);
			}
			users.put(user.userName.hashCode(), user);
		}
		
		//查询处理
		int q = scanner.nextInt();
		scanner.nextLine();
		for(int i = 0;i < q;i++) {
			String query = scanner.nextLine();
			String[] strs = query.split(" ");
			String username = strs[0];
			User user = users.get(username.hashCode());
			if(user == null) {//用户不存在
				System.out.println(false);
				continue;
			}
			Privilege privilege = new Privilege();
			if(strs[1].contains(":")) {
				String[] strss = strs[1].split(":");
				privilege.name = strss[0];
				privilege.range = Integer.parseInt(strss[1]);
			} else {
				privilege.name = strs[1];
			}
			Object obj = user.hasPrivilege(privilege);
			System.out.println(obj);
		}
		
	}
	//权限类
	public static class Privilege{
		public String name;
		public Integer range = -1;//默认-1表示不分等级 权限范围
		@Override
		public int hashCode() {
			return name.hashCode();
		}
	}
	//角色权限
	public static class RolePrivilege{
		public Privilege privilege;
		public int level = -1;
	}
	//角色类
	public static class Role{
		public String roleName;
		public int privilegeNum = 0;
		public Map<Integer,RolePrivilege> privileges = new HashMap<>();
		public void hasPrivilege(Privilege privilege, List<Object> result) {
			RolePrivilege rp = privileges.get(privilege.name.hashCode());
			//根本没有该权限
			if(rp == null) {
				result.add(false);
				return ;
			}
			//有权限但是不含等级 没有考虑如果权限本身不含等级但是查询的时候出现等级的情况
			if(rp.level == -1) {
				if(privilege.range!=-1) {//如果权限没有等级 但是查询的时候给出了等级
					result.add(false);
					return;
				}
				result.add(true);
				return;
			}
			
			//权限含等级 但是查询的时候没有写等级
			if(privilege.range==-1) {
				result.add(rp.level);
				return;
			} else if(privilege.range <=rp.level){ //权限含等级 并且查询给出等级 那么满足条件就认为是有权限的
				result.add(true);
				return;
			}
			result.add(false);
		}
	}
	public static class User{
		public String userName;
		public int roleNum;
		public Map<Integer,Role> roles = new HashMap<>();
		//用户是否有权限
		public Object hasPrivilege(Privilege privilege) {
			Set<Integer> keys = roles.keySet();
			//出现问题就出在这里 如果已经查询到的obj不为null，继续查询可能覆盖以前的结果
			//必须角色查询完才知道是不是具有该权限，除了其中的结果有返回false的
			List<Object> result = new ArrayList<>();
			for(Integer integer : keys) {
				Role role = roles.get(integer);
				role.hasPrivilege(privilege,result);
			}
			int max = -22;
			int falseCount = 0;
			for (Object object : result) {
				if(object.toString().equals("true")) {
					return true;
				} else if(object.toString().equals("false")) {
					falseCount++;
				} else {//是整数
					int a = Integer.parseInt(object.toString());
					max = a > max? a : max;
				}
			}
			//如果所有角色中都没有找到权限
			if(falseCount == result.size()) {
				return false;
			}
			//如果返回的是整数
			if(max!=-22) {
				return max;
			}
			return false;
		}
	}
}