如有错误,欢迎大神指出!!!
寒假水题系列。。。
问题 A: 别救大兵瑞恩
时间限制: 1 Sec 内存限制: 128 MB提交: 7 解决: 4
[ 提交][ 状态][ 讨论版]
题目描述
为了拯救大兵瑞恩,一群士兵相互掩护要与敌人血战到底,而你正是敌人的军师。你已经用无人机侦查数天终于摸清了状况,你估算出了每个士兵的战斗力Hi,由于他们采取相互掩护的战法,为了消灭士兵X,你将付出的代价是X的战斗力加上所有掩护X的士兵的战斗力之和(BUT死人是不能战斗的)。作为反派,你喜欢采取各个击破的手段,即每次行动杀死一个士兵,求最少需要付出多少代价,才能把这群士兵全部干掉.
输入
第一行一个整数N 代表士兵人数
接下来一行共N个整数,依次代表第i士兵的战斗力Hi
第三行一个整数M 代表相互掩护的士兵对数
接来M行,每行两个整数A,B,代表A,B相互掩护
1<=N<=1000 0<=M<=2000
0<=Hi<=10,000
1<=A,B<=N
输出
输出一个整数最小代价
样例输入
4
10 20 30 40
3
1 4
1 2
2 3
样例输出
140
解题思路:
其实就是将战力从高到低杀掉,所以需要排个序并且记录是几号士兵,杀掉的士兵用vis记录就好了。运用vector,以及两个结构体就好了。
ac code:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <list>
#include <deque>
#include <queue>
#include <iterator>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cctype>
#define si1(a) scanf("%d",&a)
#define si2(a,b) scanf("%d%d",&a,&b)
#define sd1(a) scanf("%lf",&a)
#define sd2(a,b) scanf("%lf%lf",&a,&b)
#define ss1(s) scanf("%s",s)
#define pi1(a) printf("%d\n",a)
#define pi2(a,b) printf("%d %d\n",a,b)
#define mset(a,b) memset(a,b,sizeof(a))
#define forb(i,a,b) for(int i=a;i<b;i++)
#define ford(i,a,b) for(int i=a;i<=b;i++)
#define LL long long
#define eps 1e-8
#define INF 0x3f3f3f3f
#define mod 1000000007
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
using namespace std;
int a[10005];
bool vis[10006];
struct node
{
int v,w;
node(int _v=0,int _w=0):v(_v),w(_w){}
};
vector < node > E[10006];
struct q
{
int pos,w;
}b[10005];
bool cmp(q x,q y)
{
if(x.w==y.w)
return x.pos<y.pos;
else
return x.w>y.w;
}
void add(int u,int v,int w)
{
E[u].push_back(node(v,w));
}
int dfs(int pos)
{
int cnt=0;
for(int i=0;i<E[pos].size();i++)
{
if(vis[E[pos][i].v]) continue;
cnt+=E[pos][i].w;
}
vis[pos]=true;
return cnt;
}
int main()
{
int n;
while(~scanf("%d",&n))
{
mset(vis,false);
for(int i=1;i<=n;i++)
E[i].clear();
for(int i=1;i<=n;i++)
{
si1(a[i]);
b[i].pos=i;
b[i].w=a[i];
}
int m;
sort(b+1,b+1+n,cmp);
si1(m);
for(int i=0;i<m;i++)
{
int u,v;
si2(u,v);
add(u,v,a[v]);
add(v,u,a[u]);
}
int ans=0;
for(int i=1;i<=n;i++)
{
ans+=dfs(b[i].pos)+b[i].w;
}
printf("%d\n",ans);
}
return 0;
}