CCF 201612-3 权限查询 Java

问题描述

‍　　授权 (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 未描述，因此不具有任何权限

思路说明

• 读取角色对应的权限存放在roles里
• 读取用户对应的角色，并获取角色的所有权限，存放到users
• 判断时，根据题目要求，先提取出用户对应的所有权限。

这里将无等级权限的最高权限用-1区分开。

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Java_201612_3 {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		Map<String, Map<String, Integer>> roles = new HashMap<>();//<role,<privil,max>>
		Map<String, Map<String, Integer>> users = new HashMap<>();//<user,<privil,max>>
		// 获取所有权限
		int n = scanner.nextInt();
		for (int i = 0; i < n; i++) {
			scanner.next();// 没有用
		}

		// 获取所有角色
		int roles_num = scanner.nextInt();
		for (int i = 0; i < roles_num; i++) {
			String rolename = scanner.next();
			int num = Integer.parseInt(scanner.next());
			Map<String, Integer> rolemap = new HashMap<>();
			for (int j = 0; j < num; j++) {
				String string = scanner.next();
				int index = string.indexOf(":");
				String privel = string;
				int max = -1;//-1表示无等级
				if (index > 0) {
					privel = string.substring(0, index);
					max = Integer.parseInt(string.substring(index + 1));
					max = Math.max(max, rolemap.getOrDefault(privel, 0));
				}
				rolemap.put(privel, max);
			}
			roles.put(rolename, rolemap);
		}

		// 获取所有用户
		int user_num = scanner.nextInt();
		for (int i = 0; i < user_num; i++) {
			String username = scanner.next();
			int num = Integer.parseInt(scanner.next());
			Map<String, Integer> usermap = new HashMap<>();
			for (int j = 0; j < num; j++) {
				String role = scanner.next();
				Map<String, Integer> rolemap = roles.get(role);
				for (String str : rolemap.keySet()) {
					usermap.put(str, Math.max(rolemap.get(str), usermap.getOrDefault(str, 0)));
				}
			}
			users.put(username, usermap);
		}

		// 判断
		int test_num = scanner.nextInt();
		for (int i = 0; i < test_num; i++) {
			String user = scanner.next();
			String ques = scanner.next();
			Map<String, Integer> user_privileges = users.get(user);
			if (user_privileges == null) {
				System.out.println("false");
				continue;
			}
			int index = ques.indexOf(":");
			if (index > 0) {
				String privilege = ques.substring(0, index);
				int max = Integer.parseInt(ques.substring(index + 1));
				if (user_privileges.containsKey(privilege)) {
					int real_max = user_privileges.get(privilege);
					if (real_max >= max)
						System.out.println("true");
					else
						System.out.println("false");
				} else {
					System.out.println("false");
				}
			} else {
				if (user_privileges.containsKey(ques)) {
					int real_max = user_privileges.get(ques);
					if (real_max != -1)
						System.out.println(real_max);
					else
						System.out.println("true");
				} else {
					System.out.println("false");
				}
			}
		}
		scanner.close();
	}
}