[JSOI2017]原力(分块+map(hash))

惯例的分块（第一次做实在想不到但这是一个套路，不仅在图上，在数学动规等处也有应用），度数比$\sqrt n$大的点成为大点，其余称为小点。

只有大点的三元环直接暴力即可，然后对于每个小点，枚举它的两条出边hash判断。枚举第一条边的复杂度与边数同阶，枚举第二条边的复杂度为边数成小点的度数，即不超过$O(m\sqrt n)$。

#include<map>
#include<cmath>
#include<cstdio>
#define rep(i,l,r) for (int i=l; i<=r; i++)
typedef long long ll;
using namespace std;

const int N=50100,mod=1000000007;
struct data{
int x,y,z; data() {}
data(int a,int b,int c) {x=a,y=b,z=c;}
bool operator<(const data &a)const {return x == a.x ? y == a.y ? z < a.z : y < a.y : x < a.x;}
};
map<data,ll> mp;
int n,m,si,x,y,z,t,head
,to[N << 2],val[N << 2],opt[N << 2],next[N << 2],cnt,d
,id[350],tot;
char str[5];

void add(int x,int y,int v,int c){ to[++cnt]=y,val[cnt]=v,opt[cnt]=c,next[cnt]=head[x],head[x]=cnt; }

int main(){
ll ans=0; scanf("%d%d",&n,&m),si=(int)sqrt(m);
rep(i,1,m){
scanf("%d%d%d%s",&x,&y,&z,str);
t=(str[0] == 'R' ? 1 : str[0] == 'G' ? 2 : 3);
add(x,y,z,t),add(y,x,z,t),d[x] ++,d[y] ++ ;
(mp[data(x,y,t)] += z) %= mod,(mp[data(y,x,t)] += z) %= mod;
}
rep(i,1,n) if(d[i] >= si) id[++tot]=i;
rep(i,1,tot) rep(j,1,tot) rep(k,1,tot)
ans=(ans+mp[data(id[i],id[j],1)]*mp[data(id[i],id[k],2)]%mod*mp[data(id[j],id[k],3)])%mod;
rep(i,1,n) if(d[i] < si)
for(int j=head[i] ; j ; j=next[j])
if(d[to[j]] >= si || to[j] > i)
for(int k=next[j]; k; k=next[k])
if(opt[k]!=opt[j] && (d[to[k]]>=si || to[k]>i))
ans=(ans+mp[data(to[j],to[k],6-opt[j]-opt[k])]*val[j]%mod*val[k])%mod;
printf("%lld\n",ans);
return 0;
}