CCF 角色授权 (Java题解)

一、题目

题目链接：ccf官网

二、题解（Java版）

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

//实际上组和用户没必要区分，统一看作用户名称

class Role {
	String roleName; // 角色名称
	HashSet<String> operate = new HashSet<>(); // 操作清单
	HashSet<String> resType = new HashSet<>(); // 资源种类清单
	HashSet<String> resName = new HashSet<>(); // 资源名称清单
}

public class Main {

	static Map<String, HashSet<String>> user_role = new HashMap<>(); // 一个用户对应的角色可能有多个，所以用数组存
	static List<Role> roleList = new ArrayList<>();
	static boolean flag;

	public static void main(String[] args) {
		
		Scanner in = new Scanner(System.in);
		int n= in.nextInt(); //角色数量
		int m = in.nextInt(); //角色关联数量
		int q = in.nextInt(); //待检查的操作数量
		in.nextLine();
		
		for (int i = 0; i < n; i++) {
			Role role = new Role();
			String[] data = in.nextLine().split(" ");
			role.roleName = data[0];
			int operateNum = Integer.parseInt(data[1]);
			for (int j = 0; j < operateNum; j++) {
				role.operate.add(data[j + 2]);
			}
			int resType = Integer.parseInt(data[operateNum + 2]);
			for (int j = 0; j < resType; j++) {
				role.resType.add(data[j + 3 + operateNum]);
			}
			int resName = Integer.parseInt(data[operateNum + resType + 3]);
			if(resName!=0) {
				for (int j = 0; j < resName; j++) {
					role.resName.add(data[j + 4 + operateNum + resType]);
				}
			}
			roleList.add(role);
		}
		
		
		for (int i = 0; i < m; i++) {
			String data[] = in.nextLine().split(" ");
			String roleName = data[0];
			Integer userNum = Integer.parseInt(data[1]);
			for (int j = 0; j < userNum*2; j+=2) {  //这个userName不要忘了*2，直接损失60分
				String userName = data[3 + j];
				if(user_role.get(userName)!=null) {
					user_role.get(userName).add(roleName);
				}else {
					HashSet<String> set = new HashSet<>();
					set.add(roleName);
					user_role.put(userName, set);
				}
			}
		}
		
		
		for (int i = 0; i < q; i++) {
			String data[] = in.nextLine().split(" ");
			List<String> userNameGroup = new ArrayList<>();
			userNameGroup.add(data[0]);
			int userNameNum = Integer.parseInt(data[1]);
			for (int j = 0; j <userNameNum ; j++) {
				String name = data[2 + j];
				userNameGroup.add(name);
			}
			final String operaName = data[2 + userNameNum];
			final String resType = data[3 + userNameNum];
			final String resName = data[4 + userNameNum];
			flag = false;
			userNameGroup.forEach(user->{
				HashSet<String> roles = user_role.get(user);
				if(roles!=null) {
					roles.forEach(role->{
						roleList.forEach(existRole->{
							if(role.equals(existRole.roleName)) {
								if(existRole.operate.contains(operaName) || existRole.operate.contains("*")) {
									if(existRole.resType.contains(resType) || existRole.resType.contains("*")) {
										if(existRole.resName.contains(resName) || existRole.resName.isEmpty()) {
											flag = true;
										}
									}
								}
							}
						});
					});
				}
			});
			if(flag)
				System.out.println(1);
			else {
				System.out.println(0);
			}

		}

	}

}