寒假水题之别救大兵瑞恩

如有错误，欢迎大神指出！！！

寒假水题系列。。。

问题 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;
}