Submission #1046825

---

Source Code Expand

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
typedef pair<int, int>pii;
typedef long long ll;
vector<pii>pat[100000];
int col[100000];
bool flag[100000];
bool isok;
bool cont;
ll cnt[3];
int ec;
void dfs(int node, int c)
{
	if (flag[node])
	{
		if (col[node] != c)cont = true;
		return;
	}
	flag[node] = true;
	col[node] = c;
	cnt[c]++;
	int f1 = 0, f2 = 0;
	for (int i = 0; i < pat[node].size(); i++)
	{
		if (pat[node][i].second == 1)f1++;
		else f2++;
		dfs(pat[node][i].first, (c + pat[node][i].second) % 3);
	}
	ec += pat[node].size();
	if (f1 > 0 && f2 > 0)isok = true;
}
int main()
{
	int num, way;
	scanf("%d%d", &num, &way);
	for (int i = 0; i < way; i++)
	{
		int za, zb;
		scanf("%d%d", &za, &zb);
		za--, zb--;
		pat[za].push_back(make_pair(zb, 1));
		pat[zb].push_back(make_pair(za, 2));
	}
	ll ans = 0;
	for (int i = 0; i < num; i++)
	{
		if (!flag[i])
		{
			isok = cont = false;
			cnt[0] = cnt[1] = cnt[2] = 0;
			ec = 0;
			dfs(i, 0);
			if (!isok)ans += ec / 2;
			else
			{
				if (!cont)ans += cnt[0] * cnt[1] + cnt[1] * cnt[2] + cnt[2] * cnt[0];
				else ans += (cnt[0] + cnt[1] + cnt[2])*(cnt[0] + cnt[1] + cnt[2]);
			}
		}
	}
	printf("%lld\n", ans);
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:37:27: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d", &num, &way);
                           ^
./Main.cpp:41:26: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d", &za, &zb);
                          ^

Judge Result