hihoCoder 1161 八卦的小冰 编程之美2015初赛第二场

我有一种自己水掉了这一题的嫌疑><待会研究一下官方题解。

比赛的时候脑抽写了个bfs暴力，WA到最后才想起来图可能是不联通的。

在这里直接预处理计算亲密度，按照无向图计算最后结果除以二即可，修改时只要修改变化的点对总亲密度的影响。

我觉得这么做最坏情况也是O(N^2)吧，比如一个节点连着十万个节点，每次询问都是修改这个节点的sex，plus询问十万次，但不造为啥还是没有T。o(╯□╰)o

#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<stdlib.h>
#include<vector>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<ctype.h>
#include<map>
#include<time.h>
#include<bitset>
#include<set>
#include<list>
using namespace std;
//hiho 1161

const int maxn=100010;
int T;
int N;
int M;
int Q;
int x;
int y;
int z;
int sex[maxn];
class node
{
public:
int idx;
int wei;
public:
node()
{
idx=0;
wei=0;
}
node(int id,int we)
{
idx=id;
wei=we;
}
void cle()
{
idx=0;
wei=0;
}
};
vector<node>mp[maxn];
long long ans;
void getnum()
{
ans=0;
for(int i=1;i<=N;i++)
{
for(int j=0;j<mp[i].size();j++)
{
int next=mp[i][j].idx;
//cout<<i<<" "<<next<<endl;
if(sex[i]!=sex[next])
{
ans+=mp[i][j].wei;
}
}
}
ans/=2;
}
int main()
{
freopen("input.txt","r",stdin);
//freopen("out2.txt","w",stdout);
scanf("%d",&T);
for(int ca=1;ca<=T;ca++)
{
scanf("%d %d %d",&N,&M,&Q);
memset(sex,0,sizeof(sex));
ans=0;
for(int i=0;i<=N;i++)
{
mp[i].clear();
for(int j=0;j<mp[i].size();j++)
{
mp[i][j].cle();
}
}
for(int i=1;i<=N;i++)
{
int gen;
scanf("%d",&gen);
sex[i]=gen;
}
for(int i=1;i<=M;i++)
{
scanf("%d %d %d",&x,&y,&z);
node tmp=node(y,z);
mp[x].push_back(tmp);
tmp=node(x,z);
mp[y].push_back(tmp);
// if(sex[x]!=sex[y]) it's also fine
// {
// ans+=z;
// }
}
getnum();
printf("Case #%d:\n",ca);
for(int i=1;i<=Q;i++)
{
int op;
scanf("%d",&op);
if(op==1)
{
scanf("%d",&x);
sex[x]=1-sex[x];
for(int j=0;j<mp[x].size();j++)
{
int next=mp[x][j].idx;
if(sex[x]==sex[next])
{
ans-=mp[x][j].wei;
}
else
{
ans+=mp[x][j].wei;
}
}
}
else if(op==2)
{
bool flg=false;
scanf("%d %d %d",&x,&y,&z);
for(int j=0;j<mp[x].size();j++)
{
if(mp[x][j].idx==y)
{
int pre=mp[x][j].wei;
mp[x][j].wei=z;
flg=true;
if(sex[x]!=sex[y])
{
ans+=(z-pre);
}
break;
}
}
for(int j=0;j<mp[y].size();j++)
{
if(mp[y][j].idx==x)
{
flg=true;
mp[y][j].wei=z;
break;
}
}
if(flg==false)// if not exist
{
node tmp=node(y,z);
mp[x].push_back(tmp);
tmp=node(x,z);
mp[y].push_back(tmp);
if(sex[x]!=sex[y])
{
ans+=z;
}
}
}
else if(op==3)
{
printf("%lld\n",ans);//之前RE是因为这里写成了I64d ><
}
}
}
return 0;
}