Submission #966046

---

Source Code Expand

#pragma comment(linker, "/STACK:102400000,102400000")
#include <bits/stdc++.h>
using namespace std;
#define vi vector<int>
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define all(x) x.begin(),x.end()
#define SZ(x) (int)(x.size())
#define rep(i,a,b) for(int i=a;i<b;i++)
#define per(i,a,b) for(int i=b-1;i>=a;i--)
#define inf 1000000007
#define mod 1000000007
#define x first
#define y second
#define pi acos(-1.0)
#define DBG(x) cerr<<(#x)<<"="<<x<<"\n";
//#define dprintf(...) 
#define hash _hash
#define next _next
//#define dprintf(...) fprintf(outFile,__VA_ARGS__)
 
#define FOREACH(it,x) for(__typeof(x.begin()) it=x.begin();it!=x.end();it++)
#define ull unsigned long long
#define ll long long
#define N 100005
 
template <class T,class U>inline void Max(T &a,U b){if(a<b)a=b;}
template <class T,class U>inline void Min(T &a,U b){if(a>b)a=b;}
 
//FILE* outFile;
inline void add(int &a,int b){a+=b;while(a>=mod)a-=mod;}


int pow(int a,int b){
    int ans=1;
    while(b){
        if(b&1)ans=ans*(ll)a%mod;
        a=(ll)a*a%mod;b>>=1;
    }
    return ans;
}

vector<pii>g[N];
ll c[3];
int f,e,vis[N];
void dfs(int u,int d){
    c[d]++;vis[u]=d+1;
    FOREACH(it,g[u]){
        int j=it->x,k=it->y;e++;
        if(vis[j]){
            f&=((d+k)%3==vis[j]-1);
        }
        else dfs(j,(d+k)%3);
    }
}
int main(){
    //freopen("1.txt","w",stdout);
    //cout<<setprecision(3)<<fixed;
    int T,i,j,k,n,m=0,K=0;
    scanf("%d%d",&n,&m);
    while(m--){
        scanf("%d%d",&i,&j),i--,j--;
        g[i].pb({j,1});
        g[j].pb({i,2});
    }
    ll ans=0;
    rep(i,0,n)if(!vis[i]){
        e=0;f=1;c[0]=c[1]=c[2]=0;
        dfs(i,0);
        if(f){
            if(c[1]&&c[2])ans+=c[0]*c[1]+c[1]*c[2]+c[2]*c[0];
            else ans+=e/2;
        }
        else ans+=(c[0]+c[1]+c[2])*(c[0]+c[1]+c[2]);
    }
    printf("%lld\n",ans);
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:61:24: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d",&n,&m);
                        ^
./Main.cpp:63:36: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d",&i,&j),i--,j--;
                                    ^

Judge Result