威虎山上的分配
描述
每年过年的时候,座山雕都会给兄弟们分银子,分银子之前,座山雕允许大伙儿发表意见,因为要是没法满足所有人的意见,指不定谁要搞出什么大新闻。不过每个人在提意见的时候只能说:“我认为 A 分的银子应该比 B 多!”。座山雕决定要找出一种分配方案,满足所有人的意见,同时使得所有人分得的银子总数最少,并且每个人分得的银子最少为 100两。
输入
第一行两个整数 n,m(0<n≤10000,0<m≤20000),表示总人数和总意见数;
以下 m 行,每行两个整数 a,b,之间用一个空格隔开,表示某个意见认为第 a 号小弟所分得的银两应该比第 b 号小弟多,所有小弟的编号由 1 开始。
输出
若无法找到合法方案,则输出Unhappy!(不包含引号),否则输出一个数表示最少总银两数。
输入样例 1
3 2
1 2
2 3
输出样例 1
303
这道题在原来的模板上相对来说进行了改进,稍微难一点。
首先,将我们的函数:
init()初始化
内部改造:
void init(){
memset(p,-1,sizeof§);
memset(indegree,0,sizeof(indegree));
eid=0;
ans=0;
vis=0;
}
核心就加了两行:
int topo(){
queue q;
for(int i=1;i<=n;i++){
if(indegree[i]==0){
q.push(i);
a[i]=100;
vis++;
}
}
while(!q.empty()){
int b=q.front();
q.pop();
for(int i=p[b];i!=-1;i=E[i].next){
int v=E[i].v;
indegree[v]–;
if(indegree[v]==0){
q.push(v);
a[v]=a[b]+1;
vis++;
}
}
}
}
基本就这样。
看如下动画:
代码:
#include<bits/stdc++.h>
using namespace std;
const int MAX_N=200000;
const int MAX_M=2000000;
int ans,vis;
int a[MAX_M];
int indegree[MAX_M];
struct edge{
int v,next;
int len;
}E[MAX_M];
int p[MAX_N],eid;
void init(){
memset(p,-1,sizeof(p));
memset(indegree,0,sizeof(indegree));
eid=0;
ans=0;
vis=0;
}
void insert(int u,int v){
E[eid].v=v;
E[eid].next=p[u];
p[u]=eid++;
indegree[v]++;
}
int n,m;
//priority_queue<int,vector<int>,greater<int> > q;
int topo(){
queue<int> q;
for(int i=1;i<=n;i++){
if(indegree[i]==0){
q.push(i);
a[i]=100;
vis++;
}
}
while(!q.empty()){
int b=q.front();
q.pop();
for(int i=p[b];i!=-1;i=E[i].next){
int v=E[i].v;
indegree[v]--;
if(indegree[v]==0){
q.push(v);
a[v]=a[b]+1;
vis++;
}
}
}
}
int main(){
init();
//memset(indegree,0,sizeof(indegree));
cin>>n>>m;
for(int i=0;i<m;i++){
int u,v;
cin>>u>>v;
insert(v,u);
}
topo();
for(int i=1;i<=n;i++){
ans+=a[i];
}
if(vis<n){
cout<<"Unhappy!"<<endl;
}else{
cout<<ans;
}
return 0;
}