细节题,有一点需要注意,就是vector中的erase和remove配合使用才能删除元素,详见STL中erase和remove的使用
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
#define rson rt<<1|1,m+1,r
#define lson rt<<1,l,m
using namespace std;
const int maxn=10000;
int cnt,status[maxn];
string name[maxn];
map<string,int>ids;
vector<int> depend[maxn],depend2[maxn],installed;
int get_id(string s)//为每个字符串编号
{
if(ids.count(s)) return ids[s];
name[cnt]=s;
return ids[s]=cnt++;
}
bool needed(int item)
{
for(int i=0;i<depend2[item].size();++i)
if(status[depend2[item][i]]) return true;//依赖于item软件的有安装
return false;
}
void install(int item,bool toplevel)
{
if(!status[item])//没有安装
{
for(int i=0;i<depend[item].size();i++)
install(depend[item][i],false);
cout<<" Installing "<<name[item]<<'\n';
status[item]=toplevel?1:2;//安装类型,1为显示安装
installed.push_back(item);
}
}
void remove(int item,bool toplevel){
if((toplevel||status[item]==2)&&!needed(item))//只有第一个和隐式安装并且不被依赖才能删除
{
status[item]=0;
installed.erase(remove(installed.begin(),installed.end(),item),installed.end());
cout<<" Removing "<<name[item]<<'\n';
for(int i=0;i<depend[item].size();i++)
remove(depend[item][i],false);
}
}
int main()
{
ios::sync_with_stdio(false);
string s,cmd;
while(getline(cin,s))
{
cout<<s<<'\n';
if(s[0]=='E') break;
stringstream ss(s);
ss>>cmd;
if(cmd[0] == 'L')
{
for(int i=0;i<installed.size();++i)
cout<<" "<<name[installed[i]]<<'\n';
}
else
{
int id1,id2;
string s1,s2;
ss>>s1;
id1=get_id(s1);
if(cmd[0]=='D')
{
while(ss>>s2)
{
id2=get_id(s2);
depend[id1].push_back(id2);
depend2[id2].push_back(id1);
}
}
if(cmd[0]=='I')
{
if(status[id1]) cout<<" "<<s1<<" is already installed.\n";
else install(id1,true);
}
if(cmd[0]=='R')
{
if(!status[id1]) cout<<" "<<s1<<" is not installed.\n";
else if(needed(id1)) cout<<" "<<s1<<" is still needed.\n";
else remove(id1,true);
}
}
}
return 0;
}