链接:https://ac.nowcoder.com/acm/contest/10324/C
来源:牛客网
给出一个长度为n的数组,从1开始编号。
然后给若干关系(x,y,z),表示同时挑选第x个数字 和 第y个数字时会额外贡献z,同时不选会贡献-z
求在数组中挑选数字的最大的和
数据包括两个数n,m,一个长度为n的数组a[](a[0]表示第一个数字),一个第一维长度为m的二维数组,描述m组共鸣关系。
输出一个整数表示答案。
n,m<=100000,保证答案<=33000
示例1
输入
复制
2,1,[-10,-10],[[1,2,5]]
返回值
复制
-5
选两个额外加z,选0个扣z,那选x,y的任意一个额外权值就是0,但是还有本身的权值
那先将ans减去每一组z值,再将x,y的权值加上z,那么出现一个就可以抵消,出现两个额外加z是满足题意的,最后贪心求ans最大即可。
class Solution {
public:
typedef long long ll;
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* @param n int整型
* @param m int整型
* @param a int整型vector
* @param b int整型vector<vector<>>
* @return long长整型
*/
long long wwork(int n, int m, vector<int>& a, vector<vector<int> >& b) {
// write code here
vector<long long> c(n);
int i;
ll ans=0;
for(i=0;i<n;i++)
c[i+1]=a[i];
for(i=0;i<m;i++)
{
c[b[i][0]]+=b[i][2];
c[b[i][1]]+=b[i][2];
ans-=b[i][2];
}
for(i=1;i<=n;i++)
{
ans+=max(0LL,c[i]);
}
return ans;
}
};