CSP 2016-12-3 权限查询

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% 的评测用例权限都是不分等级的，查询也都不带等级

 

思路：用map保存权限与其对应的level。用map保存角色，及其拥有的权限vector。

查寻时候遍历查询。注意有分类等级需要查询其具体等级时候，需要查出最大的等级。

code:

#include<cstdio>
#include<iostream>
#include<vector>
#include<string>
#include<map>
#include<cstring>
#include<set>
#include<cmath>
#include<queue>
#include<algorithm>
#define rep(i,j,k) for(int i=j;i<k;++i)
#define mst(a,b) memset((a),(b),sizeof(a))
#define pb push_back
#include<cstring>
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int,int> vii;
typedef pair<string,int> psi;

int num(string str){
    int ans = 0;
    rep(i,0,str.size()){
        ans = ans*10 + str[i] -'0';
    }
    return ans;
}

int main()
	{
	    //freopen("input.txt","r",stdin);
	    string str,label;
        int p,r,u,t,q,cls;
        cin >> p;
        int pos;
        map<string,int> pri;
        while(p--){
            cin >> str;
            if((pos = str.find(":")) != string::npos){
                label = str.substr(0,pos);
                cls = num(str.substr(pos+1));
                pri[label] = cls;
            }
            else{
                pri[str] = -1; // -1表示不是分类等级
            }
        }
        map<string,vector<psi>> role_pri;
        string role;
        int s;
        cin >> r;
        while(r--){
            cin >> role >> s;
            rep(i,0,s){
                cin >> str;
                if((pos = str.find(":")) != string::npos){
                    label = str.substr(0,pos);
                    cls = num(str.substr(pos+1));
                    role_pri[role].pb(make_pair(label,cls));
                }
                else{
                    role_pri[role].pb(make_pair(str,-1)); // -1表示为非分类等级
                }
            }
        }

        string cur_user;
        cin >> u;
        map<string,vector<string>> user;
        while(u--){
            cin >> cur_user >> t;
            rep(i,0,t){
                cin >> str;
                user[cur_user].pb(str);
            }
        }

        string q_user;
        cin >> q;
        while(q--){
            cin >> q_user >> str;
            if(user.find(q_user) == user.end())
                cout << "false" << endl;
            else{
                if((pos = str.find(":")) != string::npos){ //有:一定是有分类等级的并且查询该等级是否存在
                    string cur_pri = str.substr(0,pos);
                    if(pri.find(cur_pri) == pri.end()){ //不存在这个类别的权限
                        cout << "false" << endl;
                        continue;
                    }
                    vector<string> vec = user[q_user];
                    bool flag = false;
                    rep(i,0,vec.size()){
                        role = vec[i];
                        //遍历 当前角色的 权限
                        for(auto it = role_pri[role].begin(); it != role_pri[role].end(); ++it){
                            if(it->fi == cur_pri && it->se >= (num(str.substr(pos+1)))){
                                flag = true;
                            }
                            if(flag)
                                break;
                        }
                        if(flag)
                            break;
                    }
                    if(flag) cout << "true" <<endl;
                    else cout << "false" << endl;
                }
                else{
                    if(pri.find(str) == pri.end()){// 不存在这个权限类别
                        cout << "false" << endl;
                    }
                    else if(pri[str] != -1){ //表示str表示有等级的权限，需要查出其具有的等级是多少
                        string cur_pri = str;
                        vector<string> vec = user[q_user];
                        bool flag = false;
                        int c = -1;
                        rep(i,0,vec.size()){
                            role = vec[i];
                            for(auto it = role_pri[role].begin(); it != role_pri[role].end(); ++it){
                                if(it->fi == cur_pri && it->se >= 0){
                                    c = max(c,it->se);//取最大权限
                                    flag = true;
                                }
                            }
                        }
                        if(flag) cout << c << endl;
                        else cout << "false" << endl;
                    }
                    else{ //不是带等级的权限
                        string cur_pri = str;
                        vector<string> vec = user[q_user];
                        bool flag = false;
                        rep(i,0,vec.size()){
                            role = vec[i];
                            for(auto it = role_pri[role].begin(); it != role_pri[role].end(); ++it){
                                if(it->fi == cur_pri){
                                    flag = true;
                                    break;
                                }
                                if(flag)
                                    break;
                            }
                            if(flag)
                                break;
                        }
                        if(flag) cout << "true" <<endl;
                        else cout << "false" << endl;
                    }
                }
            }
        }


		return 0;
	}