結果

問題 No.2293 無向辺 2-SAT
コンテスト
ユーザー kakel-san
提出日時 2024-02-01 01:26:01
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 637 ms / 4,000 ms
コード長 5,198 bytes
コンパイル時間 9,139 ms
コンパイル使用メモリ 166,996 KB
実行使用メモリ 248,924 KB
最終ジャッジ日時 2024-09-28 10:27:11
合計ジャッジ時間 42,190 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 53
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (88 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

using System;
using static System.Console;
using System.Linq;
using System.Collections.Generic;

class Program
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    static int[][] NArr(long n) => Enumerable.Repeat(0, (int)n).Select(_ => NList).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var c = NList;
        var (n, q) = (c[0], c[1]);
        var map = NArr(q);
        WriteLine(Exp(n, q, map));
    }
    static string Exp(int n, int q, int[][] map)
    {
        var mlist = new List<int[]>();
        var ans = new List<long>();
        var all = Exp(2, n, mod);
        foreach (var li in map)
        {
            if (li[0] == 3)
            {
                ans.AddRange(Sub(n, mlist, all));
                ans.Add(all);
                mlist = new List<int[]>();
            }
            else mlist.Add(li);
        }
        ans.AddRange(Sub(n, mlist, all));
        return string.Join("\n", ans);
    }
    static long Exp(long n, long p, int mod)
    {
        long _n = n % mod;
        var _p = p;
        var result = 1L;
        if ((_p & 1) == 1) result *= _n;
        while (_p > 0)
        {
            _n = _n * _n % mod;
            _p >>= 1;
            if ((_p & 1) == 1) result = result * _n % mod;
        }
        return result;
    }
    static int mod = 998_244_353;
    static int half = 499_122_177;
    static List<long> Sub(int n, List<int[]> map, long all)
    {
        if (map.Count == 0) return new List<long>();
        var set = new HashSet<int>();
        foreach (var li in map)
        {
            set.Add(li[1]);
            set.Add(li[2]);
        }
        var list = new List<int>(set);
        list.Sort();
        var dic = new Dictionary<int, int>();
        for (var i = 0; i < list.Count; ++i) dic[list[i]] = i;
        var zeros = new HashSet<int>[list.Count];
        var ones = new HashSet<int>[list.Count];
        for (var i = 0; i < list.Count; ++i)
        {
            zeros[i] = new HashSet<int>{ i };
            ones[i] = new HashSet<int>();
        }
        var uf = new UnionFindTree(list.Count);
        var ans = new List<long>(map.Count);
        var tmp = all;
        foreach (var li in map)
        {
            var a = dic[li[1]];
            var b = dic[li[2]];
            if (uf.IsSameTree(a, b))
            {
                var r = uf.GetRoot(a);
                if (li[0] == 1)
                {
                    if (zeros[r].Contains(a) != zeros[r].Contains(b)) tmp = 0;
                }
                else
                {
                    if (zeros[r].Contains(a) == zeros[r].Contains(b)) tmp = 0;
                }
            }
            else
            {
                var asize = uf.GetSize(a);
                var bsize = uf.GetSize(b);
                var ra = uf.GetRoot(a);
                var rb = uf.GetRoot(b);
                uf.Unite(a, b);
                var v = asize > bsize ? ra : rb;
                var rev = li[0] != 1;
                if (!zeros[ra].Contains(a)) rev = !rev;
                if (!zeros[rb].Contains(b)) rev = !rev;
                if (!rev)
                {
                    foreach (var zi in zeros[v ^ ra ^ rb]) zeros[v].Add(zi);
                    foreach (var oi in ones[v ^ ra ^ rb]) ones[v].Add(oi);
                }
                else
                {
                    foreach (var zi in zeros[v ^ ra ^ rb]) ones[v].Add(zi);
                    foreach (var oi in ones[v ^ ra ^ rb]) zeros[v].Add(oi);
                }
                var r = uf.GetRoot(a);
                if (r != v)
                {
                    if (!rev)
                    {
                        zeros[r] = zeros[v];
                        ones[r] = ones[v];
                    }
                    else
                    {
                        ones[r] = zeros[v];
                        zeros[r] = ones[v];
                    }
                }
                tmp = tmp * half % mod;
            }
            ans.Add(tmp);
        }
        return ans;
    }
    class UnionFindTree
    {
        int[] roots;
        public UnionFindTree(int size)
        {
            roots = new int[size];
            for (var i = 0; i < size; ++i) roots[i] = -1;
        }
        public int GetRoot(int a)
        {
            if (roots[a] < 0) return a;
            return roots[a] = GetRoot(roots[a]);
        }
        public bool IsSameTree(int a, int b)
        {
            return GetRoot(a) == GetRoot(b);
        }
        public bool Unite(int a, int b)
        {
            var x = GetRoot(a);
            var y = GetRoot(b);
            if (x == y) return false;
            if (-roots[x] < -roots[y]) { var tmp = x; x = y; y = tmp; }
            roots[x] += roots[y];
            roots[y] = x;
            return true;
        }
        public int GetSize(int a)
        {
            return -roots[GetRoot(a)];
        }
    }
}
0